def createUser(login_session):
''' register the user and return the user id'''
newUser = User(name=login_session['username'], email=login_session[
'email'])
session.add(newUser)
session.commit()
user = session.query(User).filter_by(email=login_session['email']).one()
return user.id
def delete_film():
_id = request.form.get('id')
if _id:
film = session.query(Film).filter(Film.id == _id).delete()
session.commit()
return {"status": "0", "message": "电影 " + _id + " 删除成功"}
return {"status": "1", "message": "电影编号无效"}
def get_film_info():
film_id = request.args.get('id')
if film_id:
films = session.query(Film).filter(Film.id == film_id).all()
if films:
return films[0]
return Film()
def test_query_album():
print('\n[ TEST ] Query Album')
# Get a user
me = UserAccount.query.first()
print('[USER]', me.uri)
# search all albums of a user
query = session.query(SpotifyAlbum.name).filter(
UserResource.owner_uri == me.uri, UserResource.type == 'album',
Reference.real_uri == UserResource.real_uri,
SpotifyAlbum.uri == Reference.uri)
print('\t[SQL]', query)
results = query.all()
print('[RESULTS]', len(results))
for name in results:
print('[NAME]', name)
def test_query_track():
print('\n[ TEST ] Query Track')
# print( SpotifyTrack.query.all() )
# Get a user
me = UserAccount.query.first()
print('[USER]', me.uri)
# Search all tracks of a user
query = session.query(SpotifyTrack.name).filter(
UserResource.owner_uri == me.uri, UserResource.type == 'track',
UserResource.real_uri == Reference.real_uri,
Reference.uri == SpotifyTrack.uri)
print('\t[SQL]', query)
results = query.all()
print('[RESULTS]', len(results))
for name in results:
print('[NAME]', name)
def get_film_list():
"""
获取电影信息
:return:
"""
from utils import get_urls
func_name = inspect.stack()[0][3]
hlog.enter_func(func_name)
film_list = list()
film_objs = session.query(Film).all()
for obj in film_objs:
film_id = obj.id
hlog.var('film_id', film_id)
location_list = get_urls('Location', film_id)
people_list = get_urls('People', film_id)
specie_list = get_urls('Specie', film_id)
vehicle_list = get_urls('Vehicle', film_id)
film = {
"id": obj.id,
"title": obj.title,
"description": obj.description,
"director": obj.director,
"producer": obj.producer,
"release_date": obj.release_date,
"rt_score": obj.rt_score,
"url": obj.url,
"people": people_list,
"species": specie_list,
"locations": location_list,
"vehicles": vehicle_list
}
film_list.append(film)
hlog.info("读取电影信息成功。")
hlog.exit_func(func_name)
return film_list
def editOption(category_name, option_id):
log = str(login_session.has_key('username'))
user_id = login_session['user_id']
category = get_cat_by_name(category_name)
option = get_option_by_id(option_id)
links = get_links_by_opId(option_id)
# check if obj in db
if category and option:
if request.method == 'POST':
f = request.form
if f['name'] == '' or f['description'] == '':
warning = "You need both Research Option' and 'Description'"
return render_template('edit-option.html',
log=log,
meta_title=meta_title,
meta_desc=meta_desc,
categories=get_cats(),
category=category,
option=option,
links=links,
warning=warning)
else:
update_option(option, f)
category = session.query(Category).filter_by(
id=option.cat_id).one()
flash(option.name + ' has updated')
return redirect(
url_for('option',
category_name=category.name,
option_id=option.id))
# show option page
else:
return render_template('edit-option.html',
meta_title=meta_title,
meta_desc=meta_desc,
log=log,
categories=get_cats(),
category=category,
option=option,
links=links)
def add_film():
_id = request.form.get('id')
_title = request.form.get('title')
_description = request.form.get('description')
_director = request.form.get('director')
_producer = request.form.get('producer')
_release_date = request.form.get('release_date')
_rt_score = request.form.get('rt_score')
_url = request.form.get('url')
hlog.var('_id', _id)
hlog.var('_title', _title)
hlog.var('_description', _description)
hlog.var('_director', _director)
hlog.var('_producer', _producer)
hlog.var('_release_date', _release_date)
hlog.var('_rt_score', _rt_score)
hlog.var('_url', _url)
old_film = session.query(Film).filter(Film.id == _id).all()
if old_film:
return {"status": "1", "message": "电影编号已经存在"}
film = Film(id=_id,
title=_title,
description=_description,
director=_director,
producer=_producer,
release_date=_release_date,
rt_score=_rt_score,
url=_url)
session.add(film)
session.commit()
return {"status": "0", "message": "新增成功"}
def getUserID(email):
try:
user = session.query(User).filter_by(email=email).one()
return user.id
except:
return None
def getUserInfo(user_id):
user = session.query(User).filter_by(id=user_id).one()
return user
