model_pandas.py

"""
This part of the code comprises the model portion of model-view-controller.
It takes in a filename, which is a csv file with information about resources for queer youth in MA.
It outputs a pandas DataFrame.

@author: Louise Nielsen and Apurva Raman
         nielsenlouise@github.com, apurvaraman@github.com
"""

from pandas import *
from bokeh.plotting import ColumnDataSource as plotColumnDataSource


class Model(object):
    """A Model() object contains a pandas.DataFrame of resources.
    """

    def __init__(self, source=None, filename='test.csv', frame=None):
        self.source = source
        self.filename = filename
        if frame is None:
            self.frame = DataFrame(read_csv(self.filename))
        else:
            self.frame = frame

    def set_color(self):
        """Sets colors for each glyph based on category. Defines attributes for each glyph.
        """
        colormap = {'health': 'red',
                    'support': 'green',
                    'housing': 'blue',
                    'advocacy': 'purple',
                    'legal': 'cyan'
                    }
        self.frame['color'] = self.frame['Category'].map(lambda x: colormap[x])
        self.source = plotColumnDataSource(dict(
                                      name=self.frame['Name'],
                                      lat=self.frame['Lat'],
                                      lon=self.frame['Lon'],
                                      address=self.frame['Address'],
                                      category=self.frame['Category'],
                                      color=self.frame['color']))

    def update_model(self):
        """This is going to update the model, probably has to filter things.

        We did not have time before the due date to implement buttons and
        filtering, which was the intent of this.
        """
        pass  # TODO: Implement this.

if __name__ == '__main__':
    thing = Model()
    thing.set_color
    print thing.frame