def get_user_for_tel(telephone):
session = orm.get_orm_session()
users = session.query(orm.User).filter_by(telephone=telephone).all()
session.bind.dispose()
if users:
return users
return None
def add_user_simple(tel, tok="toke"):
session = orm.get_orm_session()
user = orm.User(tel, tok)
session.add_all([user])
session.commit()
session.bind.dispose()
return user
def get_users():
session = orm.get_orm_session()
accounts = session.query(orm.User).all()
session.bind.dispose()
for a in accounts:
print a
return accounts
def get_user_for_id(user_id):
session = orm.get_orm_session()
users = session.query(orm.User).filter_by(user_id=user_id).all()
session.bind.dispose()
if users:
return users
return None
def get_shared_accounts():
session = orm.get_orm_session()
accounts = session.query(orm.SharedAccount).all()
session.bind.dispose()
for a in accounts:
print a
return accounts
def get_account_id_from_account_number(account_number):
session = orm.get_orm_session()
account_id = session.query(orm.Account.id).filter_by(account_number=account_number).first()
session.bind.dispose()
if account_id:
print "account_id", account_id[0]
return account_id[0]
return None
def my_accounts(telephone):
session = orm.get_orm_session()
accounts = session.query(orm.Account).join(orm.User).filter(orm.User.telephone==telephone).all()
print "my accounts", accounts
session.bind.dispose()
if accounts:
return accounts
return None
def get_connection(user_id):
session = orm.get_orm_session()
connections = session.query(orm.Connection).filter_by(user_1=user_id).all()
print "connections", connections
session.bind.dispose()
if connections:
return connections
return None
def add_connection(user_1, user_2):
print "add connection method", user_1, user_2
connection = orm.Connection(user_1, user_2)
connection_back = orm.Connection(user_2, user_1)
session = orm.get_orm_session()
session.add_all([connection, connection_back])
session.commit()
session.bind.dispose()
return True
def account_amount_update(account_number, amount):
session = orm.get_orm_session()
accounts = session.query(orm.Account).filter_by(account_number=account_number).all()
if accounts:
for account in accounts:
account.amount = amount
session.commit()
session.bind.dispose()
return True
else:
session.bind.dispose()
return False
def add_user():
print "adding user"
print request.json
print dir(request.json)
telephone = request.json['telephone']
token = request.json['token']
session = orm.get_orm_session()
user = orm.User(telephone, token)
session.add_all([user])
session.commit()
session.bind.dispose()
print "useradde ", telephone, token
return "OK"
def get_users():
print "get users"
session = orm.get_orm_session()
users = session.query(orm.User).all()
to_ret = '{"meetmeup":['
for u in users:
encoded = ORMEncoder().encode(u)
print encoded
#print json.dumps(a.__dict__, skipkeys=True)
to_ret += encoded
to_ret += ","
to_ret = to_ret[:-1]
to_ret += ']}'
session.bind.dispose()
return to_ret
def add_user_to_account_real(account_number, telephone):
#TODO check that there is no duplication of rows
print "adding user to account and telephone", account_number, telephone
account_id = get_account_id_from_account_number(account_number)
users = get_user_for_tel(telephone)
if users:
print "adding", users
shared_account = orm.SharedAccount(account_id, users[0].id)
session = orm.get_orm_session()
session.add_all([shared_account])
session.commit()
session.bind.dispose()
return True
session.bind.dispose()
return False
def get_tels_for_account(account_number):
session = orm.get_orm_session()
account_id = get_account_id_from_account_number(account_number)
telephones = session.query(orm.User.telephone).join(orm.SharedAccount).filter(orm.SharedAccount.account_id==account_id).all()
"""
account_id = get_account_id_from_account_number(account_number)
if not account_id:
return None
telephones = session.query(orm.SharedAccount.telephone).filter_by(account_id=account_id).all()
"""
session.bind.dispose()
print "telephones are", telephones
to_ret = []
for tel in telephones:
to_ret.append(tel[0])
print "toretraro", to_ret
return to_ret
def add_user():
print "adding user"
print request.json
print len( request.json)
print request.json['user_id']
print request.json['extra_data']
print request.json['name']
print request.json['user_type']
print dir(request)
user_id = request.json['user_id']
extra_data = request.json['extra_data']
name = request.json['name']
img = request.json['img']
user_type = request.json['user_type']
session = orm.get_orm_session()
user = orm.User(user_id, extra_data, name, img, user_type)
session.add_all([user])
session.commit()
session.bind.dispose()
print "useradde ", user_id, name
return "OK"
def add_account():
print "adding account"
print request.json
print dir(request.json)
telephone = request.json['telephone']
name = request.json['name']
amount = request.json['amount']
amount_needed = request.json['amount_needed']
account_number = request.json['account_number']
session = orm.get_orm_session()
user_id = get_user_for_tel(telephone)[0].id
acc = orm.Account( account_number, amount_needed, name, None, user_id, amount)
session.add_all([acc])
session.commit()
acc = get_account(account_number)
shared_acc = orm.SharedAccount( acc.id, user_id)
session.add_all([shared_acc])
session.commit()
session.bind.dispose()
print "account added ", telephone, account_number
return "OK"
def get_accounts_for_tel(telephone):
print "get_account from tel", telephone
session = orm.get_orm_session()
user = get_user_for_tel(telephone)
if not user:
print "user not found returning none"
return None
for u in user:
print "u", u
print user, user[0].id
#accounts = session.query(orm.Account).join(orm.SharedAccount).join(orm.User).filter(orm.User.telephone==telephone).all()
account_ids = session.query(orm.SharedAccount.account_id).filter_by(user_id=user[0].id).all()
print "accounts_ids", account_ids
if not account_ids:
return None
accounts = session.query(orm.Account).filter(orm.Account.id.in_([x[0] for x in account_ids])).all()
session.bind.dispose()
print "accounts", accounts
to_ret = []
for acc in accounts:
to_ret.append(acc)
return to_ret
def get_account_amount(account_number):
session = orm.get_orm_session()
amount = session.query(orm.Account.amount).filter_by(account_number=account_number).first()
session.bind.dispose()
print amount
return amount[0]
