110067722: Create the model and dao for services.

This commit creates the data model for services and user_to_service.
The dao is also created to insert, get, activate, and unrestrict the service.
This commit is contained in:
Rebecca Law
2015-12-14 15:59:29 +00:00
parent 6c40e72e78
commit 4b01335703
4 changed files with 156 additions and 0 deletions

View File

@@ -0,0 +1,38 @@
from datetime import datetime
from app import db
from app.models import Service
def insert_new_service(service_name, user):
service = Service(name=service_name,
created_at=datetime.now(),
limit=1000,
active=False,
restricted=True)
add_service(service)
service.users.append(user)
db.session.commit()
return service.id
def get_service_by_id(id):
return Service.query.get(id)
def unrestrict_service(service_id):
service = get_service_by_id(service_id)
service.restricted = False
add_service(service)
def activate_service(service_id):
service = get_service_by_id(service_id)
service.active = True
add_service(service)
def add_service(service):
db.session.add(service)
db.session.commit()