forked from dineshshetty/Android-InsecureBankv2
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmodels.py
More file actions
57 lines (47 loc) · 1.61 KB
/
Copy pathmodels.py
File metadata and controls
57 lines (47 loc) · 1.61 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
import os
import base64
#from datetime import datetime
from sqlalchemy import Column, Integer, String
from database import Base, db_session
#import settings
class User(Base):
__tablename__ = 'users'
id = Column(Integer, primary_key=True)
username = Column(String(50), unique=True)
password = Column(String(50))
first_name = Column(String(50))
last_name = Column(String(50))
def __init__(self, username=None, password=None, first_name=None, last_name=None):
self.username = username
self.password = password
self.first_name = first_name
self.last_name = last_name
def __repr__(self):
return '<User %r>' % (self.username)
@property
def values(self):
return {"username" : self.username,
"first_name" : self.first_name,
"last_name" : self.last_name,
}
class Account(Base):
__tablename__ = 'accounts'
id = Column(Integer, primary_key=True)
account_number = Column(Integer, unique=True)
type = Column(String(50))
balance = Column(Integer)
user_id = Column(Integer)
user = Column(String(50))
def __init__(self, account_number=None, type=type, balance=None, user=None):
self.account_number = account_number
self.type = type
self.balance = balance
self.user = user
def __repr__(self):
return '<Account %r>' % (self.account_number)
@property
def values(self):
return {"account_number" : self.account_number,
"type" : self.type,
"balance" : self.balance,
}