X-Git-Url: https://asedeno.scripts.mit.edu/gitweb/?a=blobdiff_plain;f=bluechips%2Fmodel%2F__init__.py;h=12e28d81a99d1fe023d361bb92b711f64bb69eb3;hb=30f5cb7f4149310ce5666bb3e49a7a723677c45e;hp=cf9b5fa28c51ed816d710934d401b0c36cddb1df;hpb=4a99aedcdeeb2157ddcd524d411fabc77a313eb8;p=bluechips.git diff --git a/bluechips/model/__init__.py b/bluechips/model/__init__.py index cf9b5fa..12e28d8 100644 --- a/bluechips/model/__init__.py +++ b/bluechips/model/__init__.py @@ -5,6 +5,8 @@ from sqlalchemy import orm from bluechips.model import meta from bluechips.model import types +from datetime import datetime + def init_model(engine): """Call me before using any of the tables or classes in the model""" @@ -28,9 +30,9 @@ expenditures = sa.Table('expenditures', meta.metadata, sa.ForeignKey('users.id'), nullable=False), sa.Column('amount', types.Currency, nullable=False), sa.Column('description', sa.types.Text), - sa.Column('date', sa.types.Date, default=sa.func.now), + sa.Column('date', sa.types.Date, default=datetime.now), sa.Column('entered_time', sa.types.DateTime, - default=sa.func.now) + default=datetime.now) ) splits = sa.Table('splits', meta.metadata, @@ -59,9 +61,9 @@ transfers = sa.Table('transfers', meta.metadata, sa.ForeignKey('users.id'), nullable=False), sa.Column('amount', types.Currency, nullable=False), sa.Column('description', sa.Text, default=None), - sa.Column('date', sa.types.Date, default=sa.func.now), + sa.Column('date', sa.types.Date, default=datetime.now), sa.Column('entered_time', sa.types.DateTime, - default=sa.func.now) + default=datetime.now) ) ### ORM Classes ### @@ -73,7 +75,7 @@ class User(object): class Expenditure(object): def __repr__(self): return '' % (self.spender, - self.amount) + self.amount) class Split(object): def __repr__(self): @@ -120,6 +122,6 @@ orm.mapper(Transfer, transfers, properties={ users.c.id)) }) -__all__ = [users, expenditures, splits, subitems, transfers, - User, Expenditure, Split, Subitem, Transfer, - meta] +__all__ = ['users', 'expenditures', 'splits', 'subitems', 'transfers', + 'User', 'Expenditure', 'Split', 'Subitem', 'Transfer', + 'meta']