def getuserfriend(user_id):
dbsession = DBSession()
userfriends = dbsession.query(UserFriend).filter(UserFriend.user_id == user_id).all()
userjson = []
for userfrined in userfriends:
friend_id = userfrined.friend_id
user = dbsession.query(User).filter(User.id == friend_id).first()
task = user.__dict__
task.pop('_sa_instance_state')
task.pop('password')
userjson.append(task)
return json.dumps(userjson)
def get_users(user_id):
dbsession = DBSession()
user = dbsession.query(User).filter(User.id == user_id).one()
task = user.__dict__
task.pop('_sa_instance_state')
task.pop('password')
return json.dumps(task)
def doaddfriends():
dbsession = DBSession()
friendsjson = request.json
user_id = friendsjson.get('user_id')
friends_id = friendsjson.get('friends_id')
friend = dbsession.query(UserFriend).filter(UserFriend.user_id == user_id,
UserFriend.friend_id == friends_id).first()
if friend is None:
user = dbsession.query(User).filter(User.id == user_id).one()
friend = dbsession.query(User).filter(User.id == friends_id).one()
dbsession.close()
user.registerObserver(friend)
message = {'message': True}
return json.dumps(message)
else:
message = {'message': False}
return json.dumps(message)
def getblogbyid(blog_id):
dbsession = DBSession()
blog = dbsession.query(Blog).filter(Blog.id == blog_id).one()
user = dbsession.query(User).filter(User.id == blog.user_id).one()
print blog
task = blog.__dict__
task.pop('_sa_instance_state')
d = task.get('issueTime')
t = d.strftime("%Y-%m-%d")
task['issueTime'] = t
task['user_name'] = user.username
task['user_photo'] = user.photo
if blog.fromBlog_id > 1:
forward_blog = dbsession.query(Blog).filter(Blog.id == blog.fromBlog_id).one()
task['forward_content'] = forward_blog.content
dbsession.close()
return json.dumps(task)
def getblogimage(blog_id):
dbsession = DBSession()
blogimages = dbsession.query(BlogImage).filter(BlogImage.blog_id == blog_id).all()
blogimagejson = []
for blogimage in blogimages:
task = blogimage.__dict__
task.pop('_sa_instance_state')
blogimagejson.append(task)
return json.dumps(blogimagejson)
def upload_file():
imagejson = request.json
user_id = imagejson.get('user_id')
user_photo = imagejson.get('user_photo')
dbsession = DBSession()
photodata = base64.b64decode(user_photo)
path = os.path.join(os.path.dirname(__file__), os.path.pardir)
pathpartent = os.path.abspath(path) + '/static/'
image = open(pathpartent + str(user_id) + '.png', 'wb')
image.write(photodata)
image.close()
dbsession = DBSession()
dbsession.query(User).filter(User.id == user_id).update(
{'photo': 'http://192.168.1.112:5000/static/' + str(user_id) + '.png'})
dbsession.commit()
dbsession.close()
message = {"message": True}
return json.dumps(message)
def getanswer(blog_id):
dbSession = DBSession()
answers = dbSession.query(Answer).filter(Answer.bolg_id == blog_id).all()
ansersjson = []
for answer in answers:
task = answer.__dict__
task.pop('_sa_instance_state')
d = task.get('resTime')
t = d.strftime("%Y-%m-%d")
task['resTime'] = t
fromUser = dbSession.query(User).filter(User.id == answer.fromUser_id).first()
toUser = dbSession.query(User).filter(User.id == answer.toUser_id).first()
if toUser.id > 1:
task['toUsername'] = toUser.username
task['from_User_name'] = fromUser.username
ansersjson.append(task)
return json.dumps(ansersjson)
def dologin():
dbsession = DBSession()
userjson = request.json
username = userjson.get('username')
userpassword = userjson.get('password')
user = dbsession.query(User).filter(User.username == username, User.password == userpassword).first()
dbsession.close()
if user is None:
message = {'message': False}
return json.dumps(message)
else:
message = {'message': True, 'user_id': user.id}
return json.dumps(message)
def getfriendblog(user_id):
dbsession = DBSession()
user = dbsession.query(User).filter(User.id == user_id).one()
bloges = user.display()
print bloges
blogjson = []
for blogs in bloges:
for blog in blogs:
task = blog.__dict__
task.pop('_sa_instance_state')
d = task.get('issueTime')
t = d.strftime("%Y-%m-%d")
task['issueTime'] = t
friend_id = blog.user_id
friend = dbsession.query(User).filter(User.id == friend_id).one()
friend_name = friend.username
task['user_name'] = friend_name
task['user_photo'] = friend.photo
if blog.fromBlog_id > 1:
forward_blog = dbsession.query(Blog).filter(Blog.id == blog.fromBlog_id).one()
task['forward_content'] = forward_blog.content
blogjson.append(task)
dbsession.close()
return json.dumps(blogjson)
def doregister():
dbsession = DBSession()
userjson = request.json
username = userjson.get('username')
password = userjson.get('password')
sex = userjson.get('sex')
photo = 'http://192.168.1.112:5000/2.png'
print sex
adduser = User(username, password, sex, photo)
user = dbsession.query(User).filter(User.username == username).first()
print user
if user is not None:
message = {"message": False}
dbsession.close()
return json.dumps(message)
else:
dbsession.add(adduser)
dbsession.commit()
dbsession.close()
message = {"message": True}
return json.dumps(message)