def post(self):
from models.service import Service, ServicesHasTag
from models.tag import Tag
from app import db, app
try:
s = Service()
# tagids = ['1','2','3']
tagids = request.json.get('tagids', [])
# tag必须是体系类tag。
tags = Tag.filter_tags(tagids, '体系')
if not tags:
return '标签是必选项', 400
s.name = request.json.get('name')
s.description = request.json.get('description')
s.image = request.json.get('image')
kubernetes = request.json.get('kubernetes')
if type(kubernetes) != dict:
return '配置请输入json格式', 400
s.kubernetes = json.dumps(kubernetes)
s.devicemodel = app.config['DEVICEMODEL']
s.createdAt = datetime.datetime.now()
db.session.add(s)
db.session.flush()
# 新建service时指定tag
for t in tags:
sht = ServicesHasTag()
sht.services_id = s.id
sht.tag_id = t
db.session.add(sht)
db.session.commit()
except Exception as e:
print(e)
db.session.rollback()
return 'ERROR', 500
return 'success', 200
def make_random_service(self):
new_service = Service()
new_service.name = "Service 45"
new_service.description = "Much service"
new_service.service_type = "night"
new_service.route = "[{'destination':'Gyle Centre','points':[{'latitude':55.93879,'longitude':-3.104563},{'latitude':55.93862,'longitude':-3.105503}]}]"
return new_service
def update_services_stops():
# Delete all data first
delete_services_stops()
# Get services as json data
json_data = requests.get(API_SERVICES, headers=API_HEADER)
service_dict = json.loads(json_data.text)
# Array of service objects
services = service_dict["services"]
# Get stops as json data
json_data = requests.get(API_STOPS, headers=API_HEADER)
stop_dict = json.loads(json_data.text)
# Array of stop objects
stops = stop_dict["stops"]
# Update the tables' last_updated field
service_date = datetime.fromtimestamp(int(service_dict["last_updated"]))
update_table_date("service", service_date)
stop_date = datetime.fromtimestamp(int(stop_dict["last_updated"]))
update_table_date("stop", stop_date)
# Create a service instance of each service and store them in a new dictionary
service_dict = {}
for service in services:
new_service = Service()
new_service.name = service["name"]
new_service.service_type = service["service_type"]
new_service.description = service["description"]
new_service.route = json.dumps(service["routes"][0]["points"]) if len(service["routes"]) > 0 else "none"
service_dict[service["name"]] = new_service
db_session.add(new_service)
# Create a stop instance of each stop and add corresponding services to them
for stop in stops:
new_stop = Stop()
new_stop.name = stop["name"]
new_stop.latitude = stop["latitude"]
new_stop.longitude = stop["longitude"]
# Add services to stops
# Must take the set because sometimes the services are listed multiple times for a particular stop.
stop_services = set(stop["services"])
for service_name in stop_services:
new_stop.services.append(service_dict[service_name])
db_session.add(new_stop)
db_session.commit()
db_session.flush()
def build_sample_db():
from models.service import Service
db.drop_all()
db.create_all()
# create services
service_list = []
for tmp in [
"Beauticians", "Carpenters", "ComputerTechnicians", "Electricians",
"Gardeners", "Laborers", "Masseuses", "Painters", "Plumbers",
"Therapists"
]:
service = Service()
service.name = tmp
service_list.append(service)
db.session.add(service)
db.session.commit()
return
from models.base import db
from models.service import Service
db.drop_all()
db.create_all()
# create services
service_list = []
for tmp in [
"Beauticians", "Carpenters", "ComputerTechnicians", "Electricians",
"Gardeners", "Laborers", "Masseuses", "Painters", "Plumbers",
"Therapists"
]:
service = Service()
service.name = tmp
service_list.append(service)
db.session.add(service)
db.session.commit()
