class UserMod(db.Model):
__tablename__ = 'user_mod'
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
user_id = db.Column(db.Integer, db.ForeignKey(User.id))
type = db.Column(db.Enum(*UserModType.choices(), name='user_mod_types'))
value = db.Column(JSON, nullable=True)
user = db.relationship(User, backref=db.backref('usermods', lazy='joined'))
class ExchangeRate(db.Model):
__tablename__ = 'exchange_rate'
id = db.Column(db.Integer, primary_key=True)
create_at = db.Column(db.DateTime, default=datetime.datetime.now)
name = db.Column(db.String(255))
date_time = db.Column(db.DateTime, default=datetime.datetime.now)
source_id = db.Column(db.Integer, db.ForeignKey(Amount.id))
source = db.relationship(Amount,
foreign_keys=source_id,
backref=db.backref('source_amounts',
lazy='joined'))
compare_id = db.Column(db.Integer, db.ForeignKey(Amount.id))
compare = db.relationship(Amount,
foreign_keys=compare_id,
backref=db.backref('compare_amounts',
lazy='joined'))
description = db.Column(db.String(500), nullable=True)
class Salary(db.Model):
__tablename__ = 'salary'
id = db.Column(db.Integer, primary_key=True)
create_at = db.Column(db.DateTime, default=datetime.datetime.now)
amount_id = db.Column(db.Integer, db.ForeignKey(Amount.id))
amount = db.relationship(Amount,
foreign_keys=amount_id,
backref=db.backref('salary_amounts',
lazy='joined'))
date_time = db.Column(db.DateTime, default=datetime.datetime.now)
exchange_rate_id = db.Column(db.Integer, db.ForeignKey(ExchangeRate.id))
exchange_rate = db.relationship(ExchangeRate,
foreign_keys=exchange_rate_id,
backref=db.backref(
'exchange_rate_salaries',
lazy='joined'))
description = db.Column(db.String(500), nullable=True)
class CashFlow(db.Model):
__tablename__ = 'cash_flow'
id = db.Column(db.Integer, primary_key=True)
create_at = db.Column(db.DateTime, default=datetime.datetime.now)
name = db.Column(db.String(255))
description = db.Column(db.String(500), nullable=True)
amount_id = db.Column(db.Integer, db.ForeignKey(Amount.id))
amount = db.relationship(Amount,
backref=db.backref('cash_flow_amounts',
lazy='joined'))
class Costs(db.Model):
__tablename__ = 'costs'
id = db.Column(db.Integer, primary_key=True)
create_at = db.Column(db.DateTime, default=datetime.datetime.now)
cash_flow_id = db.Column(db.Integer, db.ForeignKey(CashFlow.id))
cash_flow = db.relationship(CashFlow,
backref=db.backref('costs_cash_flows',
lazy='joined'))
name = db.Column(db.String(255))
description = db.Column(db.String(500), nullable=True)
class Bank(db.Model):
__tablename__ = 'bank'
id = db.Column(db.Integer, primary_key=True)
create_at = db.Column(db.DateTime, default=datetime.datetime.now)
name = db.Column(db.String(255))
bank_account_id = db.Column(db.Integer, db.ForeignKey(BankAccount.id))
bank_account = db.relationship(BankAccount,
backref=db.backref('bank_bank_accounts',
lazy='joined'))
description = db.Column(db.String(500), nullable=True)
class UserProfile(db.Model):
__tablename__ = 'user_profile'
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
user_id = db.Column(db.Integer, db.ForeignKey(User.id))
phone = db.Column(db.String(17))
user = db.relationship(User,
backref=db.backref('user_profile_users',
lazy='joined'))
def __unicode__(self):
return unicode(self.id)
class Amount(db.Model):
__tablename__ = 'amount'
id = db.Column(db.Integer, primary_key=True)
create_at = db.Column(db.DateTime, default=datetime.datetime.now)
value = db.Column(db.Numeric(10, 2))
currency = db.Column(db.Enum(*AmountType.choices(), name='amount_types'),
default=AmountType.UAH)
description = db.Column(db.String(500), nullable=True)
user_id = db.Column(db.Integer, db.ForeignKey(User.id))
user = db.relationship(User,
backref=db.backref('useramounts', lazy='joined'))
def __unicode__(self):
return u'{0} {1}'.format(self.value, self.currency)
auth.models
~~~~~~~~~~~
"""
import itertools
import datetime
from sqlalchemy.dialects.postgresql import ARRAY, JSON
from flask.ext.security import UserMixin, RoleMixin, current_user
from flask_security.utils import encrypt_password
from budgetfamily import db
roles_users = db.Table(
'roles_users',
db.Column('user_id', db.Integer(), db.ForeignKey('auth_user.id')),
db.Column('role_id', db.Integer(), db.ForeignKey('role.id')),
)
class Role(db.Model, RoleMixin):
__tablename__ = 'role'
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(80), unique=True)
description = db.Column(db.Text())
permissions = db.Column(ARRAY(db.Text(), dimensions=1))
def __unicode__(self):
return self.name
