def post():
parser = reqparse.RequestParser()
parser.add_argument('login_type', type=int, required=True, help=u'登陆必须需要login_type')
parser.add_argument('user_name', type=str, required=False)
parser.add_argument('password', type=str, required=False)
parser.add_argument('open_id', type=str, required=False)
args = parser.parse_args()
login_type = args.get('login_type')
user_name = args.get('user_name', None)
password = args.get('password', None)
open_id = args.get('open_id', None)
err = {'status': 1}
if login_type == 0:
if not password:
err['message'] = '登陆用户必须有密码'
return err
if not user_name:
err['message'] = '登陆用户必须有账号名user_name，可以是登录名，可以是昵称'
return err
if User.query.filter(User.nick_name == user_name).count():
user = User.query.filter(User.nick_name == user_name).first()
elif User.query.filter(User.login_name == user_name).count():
user = User.query.filter(User.login_name == user_name).first()
else:
err['message'] = '用户不存在'
return err
if user.check_password(password):
return {'user': pickler.flatten(user), 'status': 0}
else:
err['message'] = '密码错误'
return err
if login_type == 1:
if not open_id:
err['message'] = '微博第三方登陆必须需要open_id'
return err
if User.query.filter(User.login_type == 1, User.open_id == open_id).count():
user = User.query.filter(User.login_type == 1, User.open_id == open_id)
else:
err['message'] = '不存在这个open_id'
return err
return {'user': pickler.flatten(user), 'status': 0}
if login_type == 2:
if not open_id:
err['message'] = '微博第三方登陆必须需要open_id'
return err
if User.query.filter(User.login_type == 2, User.open_id == open_id).count():
user = User.query.filter(User.login_type == 2, User.open_id == open_id)
else:
err['message'] = '不存在这个open_id'
return err
return {'user': pickler.flatten(user), 'status': 0}
def differences(obj, time_dif):
"""
计算时间差
"""
obj_pic = pickler.flatten(obj)
obj_pic['difference'] = time_dif
return obj_pic
def flatten(obj, obj2):
"""
转换json对象
"""
obj_pic = pickler.flatten(obj)
obj_pic['pic_path'] = obj2.rel_path + obj2.pic_name
return obj_pic
def get():
"""
所需参数：
无
"""
pub_types = PubType.query.filter()
resp_suc = {}
resp_suc['pub_type_list'] = []
resp_suc['pub_list'] = []
if pub_types:
for pub_type in pub_types:
pub_type_pic = pickler.flatten(pub_type)
resp_suc['pub_type_list'].append(pub_type_pic)
pub_count = Pub.query.filter(Pub.recommend == 1).count()
if pub_count > 1:
pubs = Pub.query.filter(Pub.recommend == 1)
for pub in pubs:
pub_pic = pickler.flatten(pub)
resp_suc['pub_list'].append(pub_pic)
else:
pub = Pub.query.filter(Pub.recommend == 1).first()
pub_pic = pickler.flatten(pub)
resp_suc['pub_list'].append(pub_pic)
return resp_suc
def get():
"""
读取所有酒吧
parser = reqparse.RequestParser()
parser.add_argument('')
广告，后期再加上。
"""
resp_suc = {}
resp_suc['list'] = []
# 酒吧所有类型
pub_types = PubType.query.filter()
for pub_type in pub_types:
pub_type_pic = pickler.flatten(pub_type)
resp_suc['list'].append(pub_type_pic)
resp_suc['status'] = 0
return resp_suc
def post():
parser = reqparse.RequestParser()
parser.add_argument('login_type', type=int, required=True, help=u'缺少参数 login_type必须为 0（注册用户），1（微博用户），2（QQ用户）')
parser.add_argument('nick_name', type=str, required=True, help=u'缺少参数 nick_name必须唯一')
parser.add_argument('password', type=str, required=False)
parser.add_argument('login_name', type=str, required=False)
parser.add_argument('open_id', type=str, required=False)
args = parser.parse_args()
login_type = args.get('login_type')
nick_name = args.get('nick_name')
password = args.get('password', None)
login_name = args.get('login_name', None)
open_id = args.get('open_id', None)
err = {'status': 1}
if login_type == 0: # 注意纯数字的昵称和手机号的冲突，以后不同用户的昵称和不能用户登录名不能相同，同一个用户可以
if not password:
err['message'] = '注册用户必须要有密码'
return err
if not login_name:
err['message'] = '注册用户必须有登陆名'
return err
if nick_name.count('@'):
err['message'] = 'nick_name不能包含@符号'
return err
if User.query.filter(User.login_name == login_name).count():
err['message'] = 'login_name已重复'
return err
if User.query.filter(User.nick_name == login_name).count():
err['message'] = 'login_name已重复'
return err
if User.query.filter(User.nick_name == nick_name).count():
err['message'] = 'nick_name已重复'
return err
if User.query.filter(User.login_name == nick_name).count():
err['message'] = 'nick_name已重复'
return err
user = User(login_type=login_type, nick_name=nick_name, password=password, login_name=login_name)
db.add(user)
db.commit()
user = User.query.filter(User.login_name == login_name).first()
user_info = UserInfoDb(user_id=user.id)
if login_name.count('@'):
user_info.email = login_name
db.add(user_info)
db.commit()
if login_type == 1 or login_type == 2:
if not open_id:
err['message'] = '第三方登陆用户必须有open_id'
return err
if nick_name.count('@'):
err['message'] = 'nick_name不能包含@符号'
return err
if User.query.filter(User.open_id == open_id).count():
err['message'] = 'open_id已重复'
return err
if User.query.filter(User.nick_name == nick_name).count():
err['message'] = 'nick_name已重复'
return err
user = User(login_type=login_type, nick_name=nick_name, open_id=open_id)
db.add(user)
db.commit()
user = User.query.filter(User.login_name == login_name).first()
user_info = UserInfoDb(user_id=user.id)
db.add(user_info)
db.commit()
return {'user': pickler.flatten(User.query.filter(User.nick_name == nick_name).first()), 'status': 0}
def post(user_id):
parser = reqparse.RequestParser()
parser.add_argument('login_type', type=int, required=True, help=u'登陆必须需要login_type')
parser.add_argument('password', type=str, required=False)
parser.add_argument('open_id', type=str, required=False)
parser.add_argument('new_password', type=str, required=False)
parser.add_argument('login_name', type=str, required=False)
parser.add_argument('nick_name', type=str, required=False)
parser.add_argument('system_message_time', type=int, required=False)
parser.add_argument('mobile', type=str, required=False)
parser.add_argument('tel', type=str, required=False)
parser.add_argument('real_name', type=str, required=False)
parser.add_argument('sex', type=int, required=False)
parser.add_argument('birthday_type', type=int, required=False)
parser.add_argument('birthday', type=str, required=False)
parser.add_argument('intro', type=str, required=False)
parser.add_argument('signature', type=str, required=False)
parser.add_argument('ethnicity_id', type=str, required=False)
parser.add_argument('company', type=str, required=False)
parser.add_argument('job', type=str, required=False)
parser.add_argument('email', type=str, required=False)
parser.add_argument('province_id', type=str, required=False)
parser.add_argument('city_id', type=str, required=False)
parser.add_argument('county_id', type=str, required=False)
parser.add_argument('street', type=str, required=False)
parser.add_argument('head_picture', type=werkzeug.datastructures.FileStorage, location='files')
args = parser.parse_args()
login_type = args.get('login_type')
password = args.get('password', None)
open_id = args.get('open_id', None)
new_password = args.get('new_password', None)
login_name = args.get('login_name', None)
nick_name = args.get('nick_name', None)
system_message_time = args.get('system_message_time', None)
mobile = args.get('mobile', None)
tel = args.get('tel', None)
real_name = args.get('real_name', None)
sex = args.get('sex', None)
birthday_type = args.get('birthday_type', None)
birthday = args.get('birthday', None)
intro = args.get('intro', None)
signature = args.get('signature', None)
ethnicity_id = args.get('ethnicity_id', None)
company = args.get('company', None)
job = args.get('job', None)
email = args.get('email', None)
province_id = args.get('province_id', None)
city_id = args.get('city_id', None)
county_id = args.get('county_id', None)
street = args.get('street', None)
head_picture = args.get('head_picture', None)
err = {'status': 1}
if login_type == 0:
if not password:
err['message'] = '注册用户获取和改变个人资料需要密码'
return err
user = User.query.filter(User.id == user_id).first()
if not user.check_password(password):
err['message'] = '密码错误'
return err
user_info = UserInfoDb.query.filter(UserInfoDb.user_id == user_id).first()
if login_name:
if User.query.filter(User.login_name == login_name).count():
err['message'] = 'login_name已存在'
return err
if nick_name:
if User.query.filter(User.nick_name == nick_name).count():
err['message'] = 'nick_name已存在'
return err
if head_picture:
if not allowed_file_extension(head_picture.stream.filename, HEAD_PICTURE_ALLOWED_EXTENSION):
err['message'] = '图片的格式不支持，png jpg jpeg gif支持'
return err
old_head_picture = user_info.base_path + user_info.rel_path + '/' + user_info.pic_name
user_info.upload_name = head_picture.stream.filename
user_info.base_path = HEAD_PICTURE_BASE_PATH
user_info.rel_path = HEAD_PICTURE_UPLOAD_FOLDER
user_info.pic_name = time_file_name(secure_filename(head_picture.stream.filename), sign=user_id)
head_picture.save(os.path.join(user_info.base_path+user_info.rel_path+'/', user_info.pic_name))
try:
os.remove(old_head_picture)
except:
pass
if login_name:
user.login_name = login_name
if nick_name:
user.nick_name = nick_name
if new_password:
user.change_password(password, new_password)
if system_message_time: # 1
if not user.system_message_time:
user.system_message_time = todayfstr()
else:
user.system_message_time = None
if mobile:
user_info.mobile = mobile
if tel:
user_info.tel = tel
if real_name:
user_info.real_name = real_name
if sex:
user_info.sex = sex
if birthday_type:
user_info.birthday_type = birthday_type
if birthday:
user_info.birthday = birthday
if intro:
user_info.intro = intro
if signature:
user_info.signature = signature
if ethnicity_id:
user_info.ethnicity_id = ethnicity_id
if company:
user_info.company = company
if job:
user_info.job = job
if email:
user_info.email = email
if province_id:
user_info.province_id = province_id
if city_id:
user_info.city_id = city_id
if county_id:
user_info.county_id = county_id
if street:
user_info.street = street
db.add(user)
db.add(user_info)
db.commit()
user = User.query.filter(User.id == user_id).first()
user_info = UserInfoDb.query.filter(UserInfoDb.user_id == user_id).first()
return {'user': pickler.flatten(user), 'user_info': pickler.flatten(user_info), 'status': 0}
if login_type == 1 or login_type == 2:
if not open_id:
err['message'] = '第三方登录用户必须需要open_id'
return err
user = User.query.filter(User.id == user_id).first()
user_info = UserInfoDb.query.filter(UserInfoDb.user_id == user_id).first()
if nick_name:
if User.query.filter(User.nick_name == nick_name).count():
err['message'] = 'nick_name已存在'
return err
if head_picture:
if not allowed_file_extension(head_picture.stream.filename, HEAD_PICTURE_ALLOWED_EXTENSION):
err['message'] = '图片的格式不支持，png jpg jpeg gif支持'
return err
old_head_picture = user_info.base_path + user_info.rel_path + '/' + user_info.pic_name
user_info.upload_name = head_picture.stream.filename
user_info.base_path = HEAD_PICTURE_BASE_PATH
user_info.rel_path = HEAD_PICTURE_UPLOAD_FOLDER
user_info.pic_name = time_file_name(secure_filename(head_picture.stream.filename), sign=user_id)
head_picture.save(os.path.join(user_info.base_path+user_info.rel_path+'/', user_info.pic_name))
try:
os.remove(old_head_picture)
except:
pass
if nick_name:
user.nick_name = nick_name
if system_message_time: # 1
if not user.system_message_time:
user.system_message_time = todayfstr()
else:
user.system_message_time = None
if mobile:
user_info.mobile = mobile
if tel:
user_info.tel = tel
if real_name:
user_info.real_name = real_name
if sex:
user_info.sex = sex
if birthday_type:
user_info.birthday_type = birthday_type
if birthday:
user_info.birthday = birthday
if intro:
user_info.intro = intro
if signature:
user_info.signature = signature
if ethnicity_id:
user_info.ethnicity_id = ethnicity_id
if company:
user_info.company = company
if job:
user_info.job = job
if email:
user_info.email = email
if province_id:
user_info.province_id = province_id
if city_id:
user_info.city_id = city_id
if county_id:
user_info.county_id = county_id
if street:
user_info.street = street
db.add(user)
db.add(user_info)
db.commit()
user = User.query.filter(User.id == user_id).first()
user_info = UserInfoDb.query.filter(UserInfoDb.user_id == user_id).first()
return {'user': pickler.flatten(user), 'user_info': pickler.flatten(user_info), 'status': 0}
def get():
"""
需要分类的id
参数:
type_id
"""
parser = reqparse.RequestParser()
parser.add_argument('type_id', type=str, required=True, help=u'酒吧类型type_id必须。')
args = parser.parse_args()
resp_suc = {}
resp_suc['list'] = []
resp_suc['hot_list'] = []
Session = sessionmaker()
Session.configure(bind=engine)
session = Session()
if args['type_id']:
result_count = session.query(PubPicture).\
join(Pub).\
join(PubTypeMid).\
filter(PubTypeMid.pub_type_id == int(args['type_id']), Pub.recommend == 1).\
group_by(PubPicture.pub_id).count()
if result_count > 1:
results = session.query(PubPicture). \
join(Pub). \
join(PubTypeMid).\
filter(PubTypeMid.pub_type_id == int(args['type_id']), Pub.recommend == 1).\
group_by(PubPicture.pub_id)
for result in results:
if result:
result_pic = flatten(result, result)
resp_suc['hot_list'].append(result_pic)
else:
result = session.query(PubPicture). \
join(Pub). \
join(PubTypeMid).\
filter(PubTypeMid.pub_type_id == int(args['type_id']), Pub.recommend == 1).\
group_by(PubPicture.pub_id).first()
if result:
result_pic = flatten(result, result)
resp_suc['hot_list'].append(result_pic)
pub_type_count = PubTypeMid.query.filter(PubTypeMid.pub_type_id == int(args['type_id'])).count()
if pub_type_count > 1:
pub_types = PubTypeMid.query.filter(PubTypeMid.pub_type_id == int(args['type_id']))
for pub_type in pub_types:
pub_count = Pub.query.filter(Pub.id == pub_type.pub_id).count()
if pub_count > 1:
pubs = Pub.query.filter(Pub.id == pub_type.pub_id)
for pub in pubs:
pub_picture = PubPicture.query.filter(PubPicture.pub_id == pub.id).first()
pub_pic = flatten(pub, pub_picture)
resp_suc['list'].append(pub_pic)
else:
pub = Pub.query.filter(Pub.id == pub_type.pub_id).first()
pub_picture = PubPicture.query.filter(PubPicture.pub_id == pub.id).first()
pub_pic = flatten(pub, pub_picture)
resp_suc['list'].append(pub_pic)
else:
pub_type = PubTypeMid.query.filter(PubTypeMid.pub_type_id == int(args['type_id'])).first()
pub = Pub.query.filter(Pub.id == pub_type.pub_id).first()
pub_picture = PubPicture.query.filter(PubPicture.pub_id == pub.id).first()
pub_pic = pickler.flatten(pub)
pub_pic['pic_path'] = pub_picture.rel_path + pub_picture.pic_name
resp_suc['list'].append(pub_pic)
resp_suc['status'] = 0
return resp_suc
else:
resp_suc['message'] = 'error'
resp_suc['status'] = 1
return resp_suc