def create_entity(entity_request, user_id):
entity = Entity(name=entity_request.name,
description=entity_request.description,
user_id=user_id)
session.add(entity)
session.commit()
return entity
def setUp(self):
self.tester = app.test_client()
self.user = User(first_name='Testy',
last_name='McTesterson',
username='******',
password='******',
email='*****@*****.**')
session.add(self.user)
session.commit()
self.entity = Entity(name='My Entity',
description='My favorite entity',
user_id=self.user.id)
session.add(self.entity)
session.commit()
self.revenue = Revenue(name='Revenue 1',
description='My First Revenue',
value=1,
entity_id=self.entity.id)
self.cost = Cost(name='Cost 1',
description='My First Cost',
value=1,
entity_id=self.entity.id)
self.opex = OperatingExpense(name='Opex 1',
description='My First Opex',
value=1,
entity_id=self.entity.id)
session.add(self.revenue)
session.add(self.cost)
session.add(self.opex)
session.commit()
def create_kdbaccounts(id):
data = request.get_json()
if not data:
return bad_request('Need to post data')
message = ''
if 'username' not in data or not data.get('username', None):
message = 'Username invalid'
if 'password' not in data or not data.get('password', None):
message = 'Password invalid'
if message != '':
return bad_request(message)
for acc in session.query(KDBAccount).filter(
KDBAccount.owner_id == id).all():
print(acc.to_dict())
if acc.to_dict()['username'] == str(data['username']):
return jsonify({'message': 'Account exists!', 'code': 70000})
kdbacc = KDBAccount()
data['owner_id'] = id
kdbacc.from_dict(data)
# kdbacc.user_id += [user]
session.add(kdbacc)
session.commit()
response = jsonify({'data': kdbacc.to_dict(), 'code': 20000})
response.status_code = 201
response.headers['Location'] = url_for('api.get_user', id=id)
return response
def create_cost(cost_request, entity_id):
cost = Cost(name=cost_request.name,
description=cost_request.description,
value=cost_request.value,
entity_id=entity_id)
session.add(cost)
session.commit()
return cost
def create_revenue(revenue_request, entity_id):
revenue = Revenue(unit_name=revenue_request.unitName,
unit_description=revenue_request.unitDescription,
unit_cost=revenue_request.unitCost,
unit_count=revenue_request.unitCount,
entity_id=entity_id)
session.add(revenue)
session.commit()
return revenue
def create_operatingexpense(operatingexpense_request, entity_id):
operatingexpense = OperatingExpense(
name=operatingexpense_request.name,
description=operatingexpense_request.description,
value=operatingexpense_request.value,
entity_id=entity_id)
session.add(operatingexpense)
session.commit()
return operatingexpense
def create_user(register_request):
user = User(username=register_request.username,
first_name=register_request.firstName,
last_name=register_request.lastName,
email=register_request.email)
user.hash_password(register_request.password)
session.add(user)
session.commit()
return user
def setUp(self):
self.tester = app.test_client()
self.user = User(first_name='Testy',
last_name='McTesterson',
username='******',
password='******',
email='*****@*****.**')
session.add(self.user)
session.commit()
self.entity = Entity(user_id=self.user.id,
name='Test Entity 1',
description='This is a test entity')
session.add(self.entity)
session.commit()
def create_tbaccounts(id):
data = request.get_json()
data['owner_id'] = id
if not data:
return bad_request('Need to post data')
message = ''
if 'username' not in data or not data.get('username', None):
message = 'Username invalid'
if 'password' not in data or not data.get('password', None):
message = 'Password invalid'
if session.query(TaobaoAccount).filter(TaobaoAccount.username==data.get('username', None)).first():
message = 'Username exists'
if message != '':
return bad_request(message)
tbacc = TaobaoAccount()
tbacc.from_dict(data)
session.add(tbacc)
session.commit()
response = jsonify({'data':tbacc.to_dict(), 'code': 20000})
response.status_code = 201
response.headers['Location'] = url_for('api.get_user', id=tbacc.owner_id)
return response
def ping(self):
self.last_seen = datetime.utcnow()
session.add(self)
