拍爱

views.py 35KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086
  1. # -*- coding: utf-8 -*-
  2. from __future__ import division
  3. import os
  4. import random
  5. import records
  6. import shortuuid
  7. from curtail_uuid import CurtailUUID
  8. from django.conf import settings
  9. from django.core.files.storage import default_storage
  10. from django.db import connection, transaction
  11. from django.shortcuts import render
  12. from logit import logit
  13. from rest_framework import viewsets
  14. from TimeConvert import TimeConvert as tc
  15. from account.models import UserInfo
  16. from group.models import GroupInfo, GroupPhotoInfo, GroupUserInfo, PhotoCommentInfo, PhotoThumbUpInfo
  17. from group.serializers import GroupInfoSerializer, GroupPhotoInfoSerializer, GroupUserInfoSerializer
  18. from message.models import UserMessageInfo
  19. from utils.error.errno_utils import GroupPhotoStatusCode, GroupStatusCode, GroupUserStatusCode, UserStatusCode
  20. from utils.error.response_utils import response
  21. from utils.page_utils import pagination
  22. from utils.redis.rgroup import (del_group_photo_thumbup_flag, get_group_info, get_group_photo_comment_list,
  23. get_group_photo_data, get_group_photo_thumbup_flag, get_group_photo_thumbup_list,
  24. get_group_photo_watchers, get_group_users_info, set_group_info,
  25. set_group_photo_comment_list, set_group_photo_data, set_group_photo_thumbup_flag,
  26. set_group_photo_thumbup_list, set_group_users_info)
  27. from utils.redis.rkeys import (GROUP_LAST_PHOTO_PK, GROUP_PHOTO_WATCHER_SET, GROUP_USERS_APPLYING_SET,
  28. GROUP_USERS_DELETED_SET, GROUP_USERS_PASSED_SET, GROUP_USERS_QUIT_SET,
  29. GROUP_USERS_REFUSED_SET, LENSMAN_PHOTO_HAGGLE_TIMES, LENSMAN_PHOTO_PRICE)
  30. from utils.redis.rorder import get_lensman_order_record
  31. from utils.sql.raw import PAI2_HOME_API
  32. from utils.thumbnail_utils import make_thumbnail
  33. from utils.time_utils import origin_expired_stamps
  34. from utils.url_utils import img_url, share_url
  35. # db = records.Database(settings.DATABASE_URL['default'])
  36. r = settings.REDIS_CACHE
  37. @logit
  38. @transaction.atomic
  39. def group_create_api(request):
  40. """
  41. 群组创建
  42. :param request:
  43. :return:
  44. """
  45. user_id = request.POST.get('user_id', '')
  46. group_name = request.POST.get('group_name', '')
  47. group_default_avatar = int(request.POST.get('group_default_avatar', 0))
  48. # 用户校验
  49. try:
  50. user = UserInfo.objects.get(user_id=user_id)
  51. except UserInfo.DoesNotExist:
  52. return response(UserStatusCode.USER_NOT_FOUND)
  53. # 群组唯一标识
  54. group_id = CurtailUUID.uuid(GroupInfo, 'group_id')
  55. # 群组记录创建
  56. group = GroupInfo.objects.create(
  57. group_id=group_id,
  58. admin_id=user_id,
  59. group_name=group_name,
  60. group_default_avatar=group_default_avatar,
  61. group_from=GroupInfo.APP_GROUP,
  62. )
  63. # Redis 群组数据缓存
  64. group_info = set_group_info(group)
  65. # 群组用户记录创建
  66. GroupUserInfo.objects.create(
  67. group_id=group_id,
  68. user_id=user_id,
  69. nickname=user.final_nickname,
  70. avatar=user.avatar,
  71. admin=True,
  72. user_status=GroupUserInfo.PASSED,
  73. passed_at=tc.utc_datetime(),
  74. )
  75. # Redis 群组用户数据缓存
  76. group_users = set_group_users_info(group)
  77. # Redis 群组通过集合缓存
  78. r.sadd(GROUP_USERS_PASSED_SET % group_id, user_id)
  79. return response(200, 'Create Group Success', u'群组创建成功', {
  80. 'group_id': group_id,
  81. 'group': group_info,
  82. 'users': group_users,
  83. })
  84. @logit
  85. def group_detail_api(request):
  86. """
  87. 群组详情
  88. :param request:
  89. :return:
  90. """
  91. group_id = request.POST.get('group_id', '')
  92. user_id = request.POST.get('user_id', '')
  93. return response(200, 'Get Group Detail Info Success', u'获取群组详情成功', {
  94. 'group_id': group_id,
  95. 'group': get_group_info(group_id),
  96. 'users': get_group_users_info(group_id, user_id),
  97. })
  98. @logit
  99. def group_update_api(request):
  100. """
  101. 群组更新
  102. :param request:
  103. :return:
  104. """
  105. group_id = request.POST.get('group_id', '')
  106. admin_id = request.POST.get('admin_id', '') or request.POST.get('user_id', '')
  107. group_name = request.POST.get('group_name', '')
  108. group_desc = request.POST.get('group_desc', '')
  109. group_avatar = request.FILES.get('group_avatar', '')
  110. # 群组校验
  111. try:
  112. group = GroupInfo.objects.get(group_id=group_id)
  113. except GroupInfo.DoesNotExist:
  114. return response(GroupStatusCode.GROUP_NOT_FOUND)
  115. # 权限校验
  116. if group.admin_id != admin_id:
  117. return response(GroupStatusCode.NO_UPDATE_PERMISSION)
  118. # 群组名称更新
  119. if group_name:
  120. group.group_name = group_name
  121. # 群组描述更新
  122. if group_desc:
  123. group.group_desc = group_desc
  124. # 群组头像更新
  125. if group_avatar:
  126. _, extension = os.path.splitext(group_avatar.name)
  127. group_avatar_path = 'group/{uuid}_{extension}'.format(uuid=shortuuid.uuid(), extension=extension)
  128. if default_storage.exists(group_avatar_path):
  129. default_storage.delete(group_avatar_path)
  130. default_storage.save(group_avatar_path, group_avatar)
  131. group.group_avatar = group_avatar_path
  132. group.save()
  133. # Redis 群组数据缓存更新
  134. group_info = set_group_info(group)
  135. return response(200, 'Update Group Success', u'群组更新成功', {
  136. 'group_id': group_id,
  137. 'group': group_info,
  138. 'users': get_group_users_info(group_id, admin_id),
  139. })
  140. @logit
  141. def group_list_api(request):
  142. """
  143. 群组列表
  144. :param request:
  145. :return:
  146. """
  147. user_id = request.POST.get('user_id', '')
  148. page = int(request.POST.get('page', 1))
  149. num = int(request.POST.get('num', settings.GROUP_PER_PAGE))
  150. group_users = GroupUserInfo.objects.filter(user_id=user_id, user_status=GroupUserInfo.PASSED).order_by('-pk')
  151. group_users, left = pagination(group_users, page, num)
  152. groups = []
  153. for group_user in group_users:
  154. group_info = get_group_info(group_user.group_id)
  155. groups.append(group_info) if group_info else None
  156. return response(200, 'Get Group List Success', u'获取群组列表成功', {
  157. 'groups': groups,
  158. 'left': left,
  159. })
  160. @logit
  161. def group_join_api(request):
  162. """
  163. 申请加群
  164. :param request:
  165. :return:
  166. """
  167. group_id = request.POST.get('group_id', '')
  168. user_id = request.POST.get('user_id', '')
  169. nickname = request.POST.get('nickname', '')
  170. # 用户校验
  171. try:
  172. user = UserInfo.objects.get(user_id=user_id)
  173. except UserInfo.DoesNotExist:
  174. return response(UserStatusCode.USER_NOT_FOUND)
  175. # 群组校验
  176. try:
  177. group = GroupInfo.objects.get(group_id=group_id)
  178. except GroupInfo.DoesNotExist:
  179. return response(GroupStatusCode.GROUP_NOT_FOUND)
  180. # 群组锁定校验
  181. if group.group_lock:
  182. return response(GroupStatusCode.GROUP_HAS_LOCKED)
  183. # 群组用户记录创建,若记录不存在,则创建,若记录已存在,则更新
  184. group_user, created = GroupUserInfo.objects.get_or_create(
  185. group_id=group_id,
  186. user_id=user_id,
  187. )
  188. if group_user.user_status != GroupUserInfo.PASSED:
  189. group_user.current_id = int(r.get(GROUP_LAST_PHOTO_PK % group_id) or -1)
  190. group_user.nickname = nickname or user.final_nickname
  191. group_user.avatar = user.avatar
  192. # group_user.admin = False # Admin Field Default False, Should Not Assign
  193. group_user.user_status = GroupUserInfo.PASSED
  194. group_user.passed_at = tc.utc_datetime()
  195. group_user.save()
  196. # Redis 群组用户数据缓存
  197. set_group_users_info(group)
  198. # Redis 群组通过集合缓存
  199. r.srem(GROUP_USERS_REFUSED_SET % group_id, user_id)
  200. r.srem(GROUP_USERS_DELETED_SET % group_id, user_id)
  201. r.srem(GROUP_USERS_QUIT_SET % group_id, user_id)
  202. r.sadd(GROUP_USERS_PASSED_SET % group_id, user_id)
  203. return response(200, 'Apply Success', u'申请成功', {
  204. 'current_id': group_user.current_id,
  205. 'photos': [],
  206. 'group_id': group_id,
  207. 'group': get_group_info(group_id),
  208. 'user_id': user_id,
  209. 'users': get_group_users_info(group_id, user_id),
  210. })
  211. @logit
  212. def group_lock_api(request):
  213. """
  214. 群组锁定
  215. :param request:
  216. :return:
  217. """
  218. group_id = request.POST.get('group_id', '')
  219. admin_id = request.POST.get('admin_id', '') or request.POST.get('user_id', '')
  220. # 群组校验
  221. try:
  222. group = GroupInfo.objects.get(group_id=group_id)
  223. except GroupInfo.DoesNotExist:
  224. return response(GroupStatusCode.GROUP_NOT_FOUND)
  225. # 权限校验
  226. if group.admin_id != admin_id:
  227. return response(GroupStatusCode.NO_LOCK_PERMISSION)
  228. # 群组锁定
  229. group.group_lock = True
  230. group.save()
  231. # Redis 群组数据缓存更新
  232. set_group_info(group)
  233. return response(200, 'Lock Success', u'锁定成功')
  234. @logit
  235. def group_unlock_api(request):
  236. """
  237. 群组解锁
  238. :param request:
  239. :return:
  240. """
  241. group_id = request.POST.get('group_id', '')
  242. admin_id = request.POST.get('admin_id', '') or request.POST.get('user_id', '')
  243. # 群组校验
  244. try:
  245. group = GroupInfo.objects.get(group_id=group_id)
  246. except GroupInfo.DoesNotExist:
  247. return response(GroupStatusCode.GROUP_NOT_FOUND)
  248. # 权限校验
  249. if group.admin_id != admin_id:
  250. return response(GroupStatusCode.NO_UNLOCK_PERMISSION)
  251. # 群组解锁
  252. group.group_lock = False
  253. group.save()
  254. # Redis 群组数据缓存更新
  255. set_group_info(group)
  256. return response(200, u'Unlock Success', u'解锁成功')
  257. @logit
  258. def group_remove_api(request):
  259. """
  260. 成员移除
  261. :param request:
  262. :return:
  263. """
  264. group_id = request.POST.get('group_id', '')
  265. admin_id = request.POST.get('admin_id', '')
  266. user_id = request.POST.get('user_id', '')
  267. # 群组校验
  268. try:
  269. group = GroupInfo.objects.get(group_id=group_id)
  270. except GroupInfo.DoesNotExist:
  271. return response(GroupStatusCode.GROUP_NOT_FOUND)
  272. # 权限校验
  273. if group.admin_id != admin_id or group.admin_id == user_id: # 管理员也不允许将自己移除
  274. return response(GroupStatusCode.NO_REMOVE_PERMISSION)
  275. # 群组用户校验
  276. try:
  277. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
  278. except GroupUserInfo.DoesNotExist:
  279. return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
  280. # 群组用户移除
  281. group_user.user_status = GroupUserInfo.DELETED
  282. group_user.deleted_at = tc.utc_datetime()
  283. group_user.save()
  284. # Redis 群组数据缓存更新
  285. group_users = set_group_users_info(group)
  286. # Redis 群组删除集合缓存
  287. r.srem(GROUP_USERS_PASSED_SET % group_id, user_id)
  288. r.sadd(GROUP_USERS_DELETED_SET % group_id, user_id)
  289. return response(200, 'Remove Success', u'用户移除成功', {
  290. 'group_id': group_id,
  291. 'users': group_users,
  292. })
  293. @logit
  294. def group_quit_api(request):
  295. """
  296. 成员退出
  297. :param request:
  298. :return:
  299. """
  300. group_id = request.POST.get('group_id', '')
  301. user_id = request.POST.get('user_id', '')
  302. # 群组校验
  303. try:
  304. group = GroupInfo.objects.get(group_id=group_id)
  305. except GroupInfo.DoesNotExist:
  306. return response(GroupStatusCode.GROUP_NOT_FOUND)
  307. # 权限校验
  308. if group.admin_id == user_id: # 管理员也不允许自己退出
  309. return response(GroupStatusCode.NO_QUIT_PERMISSION)
  310. # 群组用户校验
  311. try:
  312. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
  313. except GroupUserInfo.DoesNotExist:
  314. return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
  315. # 群组用户移除
  316. group_user.user_status = GroupUserInfo.QUIT
  317. group_user.quit_at = tc.utc_datetime()
  318. group_user.save()
  319. # Redis 群组数据缓存更新
  320. group_users = set_group_info(group)
  321. # Redis 群组删除集合缓存
  322. r.srem(GROUP_USERS_PASSED_SET % group_id, user_id)
  323. r.sadd(GROUP_USERS_QUIT_SET % group_id, user_id)
  324. return response(200, 'Quit Success', u'用户退出成功', {
  325. 'group_id': group_id,
  326. 'users': group_users,
  327. })
  328. @logit
  329. def group_pass_api(request):
  330. """
  331. 申请通过
  332. :param request:
  333. :return:
  334. """
  335. group_id = request.POST.get('group_id', '')
  336. admin_id = request.POST.get('admin_id', '')
  337. user_id = request.POST.get('user_id', '')
  338. # 群组校验
  339. try:
  340. group = GroupInfo.objects.get(group_id=group_id)
  341. except GroupInfo.DoesNotExist:
  342. return response(GroupStatusCode.GROUP_NOT_FOUND)
  343. # 权限校验
  344. if group.admin_id != admin_id:
  345. return response(GroupStatusCode.NO_PASS_PERMISSION)
  346. # 群组用户校验
  347. try:
  348. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.APPLYING)
  349. except GroupUserInfo.DoesNotExist:
  350. return response(GroupStatusCode.JOIN_REQUEST_NOT_FOUND)
  351. # 群组用户通过
  352. group_user.user_status = GroupUserInfo.PASSED
  353. group_user.passed_at = tc.utc_datetime()
  354. group_user.save()
  355. # Redis 群组数据缓存更新
  356. group_users = set_group_info(group)
  357. # Redis 群组通过集合缓存
  358. r.srem(GROUP_USERS_APPLYING_SET % group_id, user_id)
  359. r.sadd(GROUP_USERS_PASSED_SET % group_id, user_id)
  360. return response(200, 'Pass Success', u'申请通过成功', {
  361. 'group_id': group_id,
  362. 'users': group_users,
  363. })
  364. @logit
  365. def group_refuse_api(request):
  366. """
  367. 申请拒绝
  368. :param request:
  369. :return:
  370. """
  371. group_id = request.POST.get('group_id', '')
  372. admin_id = request.POST.get('admin_id', '')
  373. user_id = request.POST.get('user_id', '')
  374. # 群组校验
  375. try:
  376. group = GroupInfo.objects.get(group_id=group_id)
  377. except GroupInfo.DoesNotExist:
  378. return response(GroupStatusCode.GROUP_NOT_FOUND)
  379. # 权限校验
  380. if group.admin_id != admin_id:
  381. return response(GroupStatusCode.NO_REFUSE_PERMISSION)
  382. # 群组用户校验
  383. try:
  384. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.APPLYING)
  385. except GroupUserInfo.DoesNotExist:
  386. return response(GroupStatusCode.JOIN_REQUEST_NOT_FOUND)
  387. # 群组用户拒绝
  388. group_user.user_status = GroupUserInfo.REFUSED
  389. group_user.refused_at = tc.utc_datetime()
  390. group_user.save()
  391. # Redis 群组数据缓存更新
  392. group_users = set_group_info(group)
  393. # Redis 群组拒绝集合缓存
  394. r.srem(GROUP_USERS_APPLYING_SET % group_id, user_id)
  395. r.sadd(GROUP_USERS_REFUSED_SET % group_id, user_id)
  396. return response(200, 'Refuse Success', u'申请拒绝成功', {
  397. 'group_id': group_id,
  398. 'users': group_users,
  399. })
  400. @logit
  401. def group_data_api(request):
  402. """
  403. 群组数据, 评论数, 点赞数
  404. :param request:
  405. :return:
  406. """
  407. group_id = request.POST.get('group_id', '')
  408. return response(200, 'Get Group Data Success', u'获取群组数据成功', {
  409. 'photo_datas': get_group_photo_data(group_id),
  410. })
  411. @logit
  412. def flyimg_upload_api(request):
  413. """
  414. 飞图上传/飞图列表
  415. :param request:
  416. :return:
  417. """
  418. group_id = request.POST.get('group_id', '')
  419. user_id = request.POST.get('user_id', '')
  420. nickname = request.POST.get('nickname', '')
  421. photo = request.FILES.get('photo', '')
  422. current_id = int(request.POST.get('current_id', -1))
  423. # 用户校验
  424. try:
  425. user = UserInfo.objects.get(user_id=user_id)
  426. except UserInfo.DoesNotExist:
  427. return response(UserStatusCode.USER_NOT_FOUND)
  428. # 群组用户校验
  429. try:
  430. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
  431. except GroupUserInfo.DoesNotExist:
  432. return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
  433. if photo:
  434. photo_path = 'fly/{uuid}{extension}'.format(uuid=shortuuid.uuid(), extension=os.path.splitext(photo.name)[1] or 'jpeg')
  435. photo_thumbnail_path = photo_path.replace('.', '_thumbnail.')
  436. photo_thumbnail2_path = photo_path.replace('.', '_thumbnail2.')
  437. if default_storage.exists(photo_path):
  438. default_storage.delete(photo_path)
  439. default_storage.save(photo_path, photo)
  440. # if default_storage.exists(photo_thumbnail_path):
  441. # default_storage.delete(photo_thumbnail_path)
  442. # default_storage.save(photo_thumbnail_path, photo)
  443. # 群组照片缩略图生成
  444. # 双列: 540, 40-50K
  445. photo_w, photo_h, photo_thumbnail_w, photo_thumbnail_h = make_thumbnail(
  446. os.path.join(settings.MEDIA_ROOT, photo_path).replace('\\', '/'),
  447. os.path.join(settings.MEDIA_ROOT, photo_thumbnail_path).replace('\\', '/'),
  448. settings.THUMBNAIL_MAX_WIDTH
  449. )
  450. # 单列: 1080, xx-100K
  451. photo_w, photo_h, photo_thumbnail2_w, photo_thumbnail2_h = make_thumbnail(
  452. os.path.join(settings.MEDIA_ROOT, photo_path).replace('\\', '/'),
  453. os.path.join(settings.MEDIA_ROOT, photo_thumbnail2_path).replace('\\', '/'),
  454. settings.THUMBNAIL_MAX_WIDTH2
  455. )
  456. # 群组照片记录创建
  457. group_photo = GroupPhotoInfo.objects.create(
  458. group_id=group_id,
  459. user_id=user_id,
  460. nickname=nickname or user.final_nickname,
  461. avatar=user.avatar,
  462. photo_path=photo_path,
  463. photo_w=photo_w,
  464. photo_h=photo_h,
  465. photo_thumbnail_path=photo_thumbnail_path,
  466. photo_thumbnail_w=photo_thumbnail_w,
  467. photo_thumbnail_h=photo_thumbnail_h,
  468. photo_thumbnail2_path=photo_thumbnail2_path,
  469. photo_thumbnail2_w=photo_thumbnail2_w,
  470. photo_thumbnail2_h=photo_thumbnail2_h,
  471. )
  472. # 设置群组最后一张照片PK
  473. r.set(GROUP_LAST_PHOTO_PK % group_id, group_photo.pk)
  474. # 获取从 current_id 到 now 的群组照片列表
  475. group_photos = GroupPhotoInfo.objects.filter(
  476. group_id=group_id,
  477. status=True,
  478. pk__gt=max(current_id, group_user.current_id),
  479. ).order_by(
  480. '-pk'
  481. )
  482. latest_photo = group_photos.first()
  483. return response(200, 'Flyimg Upload Success', u'飞图上传成功', {
  484. 'current_id': latest_photo and latest_photo.pk or current_id,
  485. 'photos': [photo.photo_info(user_id) for photo in group_photos],
  486. })
  487. @logit
  488. def comment_submit_api(request):
  489. """
  490. 飞图评论提交/飞图评论列表
  491. :param request:
  492. :return:
  493. """
  494. group_id = request.POST.get('group_id', '')
  495. user_id = request.POST.get('user_id', '')
  496. to_uid = request.POST.get('to_uid', '')
  497. photo_id = request.POST.get('photo_id', '')
  498. comment = request.POST.get('comment', '')
  499. # 群组用户校验
  500. try:
  501. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
  502. except GroupUserInfo.DoesNotExist:
  503. return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
  504. # 群组照片校验
  505. try:
  506. group_photo = GroupPhotoInfo.objects.get(pk=photo_id)
  507. except GroupPhotoInfo.DoesNotExist:
  508. return response(GroupPhotoStatusCode.GROUP_PHOTO_NOT_FOUND)
  509. if comment:
  510. # 群组照片评论记录创建
  511. PhotoCommentInfo.objects.create(
  512. photo_id=photo_id,
  513. user_id=user_id,
  514. nickname=group_user.nickname,
  515. avatar=group_user.avatar,
  516. to_uid=to_uid,
  517. comment=comment,
  518. )
  519. # 群组照片评论数更新
  520. group_photo.comment_num += 1
  521. group_photo.save()
  522. # Redis 群组照片数据缓存
  523. set_group_photo_data(group_id)
  524. # Redis 群组照片评论列表缓存刷新
  525. set_group_photo_comment_list(photo_id)
  526. r.sadd(GROUP_PHOTO_WATCHER_SET % photo_id, user_id)
  527. # 判断群组照片发布者是否已经被管理员移除/主动退出,如若移除/退出,则不给发布者提醒
  528. # 照片所有者(评论/点赞)不给自己提醒
  529. if r.sismember(GROUP_USERS_PASSED_SET % group_photo.group_id, group_photo.user_id) and group_photo.user_id != user_id:
  530. UserMessageInfo.objects.create(
  531. from_uid=user_id,
  532. from_nickname=group_user.nickname,
  533. from_avatar=group_user.avatar,
  534. to_uid=group_photo.user_id,
  535. group_id=group_photo.group_id,
  536. photo_id=group_photo.pk,
  537. msg_type=UserMessageInfo.COMMENT,
  538. msg_title=u'评论',
  539. msg_content=comment,
  540. )
  541. # 给所有关注者(评论/点赞)发送提醒
  542. watchers = get_group_photo_watchers(photo_id)
  543. # 从关注者中移除该(评论/点赞)者
  544. watchers.discard(user_id)
  545. # 从关注者中移除该照片所有者
  546. watchers.discard(group_photo.user_id)
  547. for watcher in watchers:
  548. UserMessageInfo.objects.create(
  549. from_uid=user_id,
  550. from_nickname=group_user.nickname,
  551. from_avatar=group_user.avatar,
  552. to_uid=watcher,
  553. group_id=group_photo.group_id,
  554. photo_id=group_photo.pk,
  555. msg_type=UserMessageInfo.COMMENT,
  556. msg_title=u'评论',
  557. msg_content=comment,
  558. )
  559. return response(200, 'Comment Success', u'评论成功', {
  560. 'comments': get_group_photo_comment_list(photo_id),
  561. })
  562. @logit
  563. def thumbup_submit_api(request):
  564. """
  565. 飞图点赞提交
  566. :param request:
  567. :return:
  568. """
  569. group_id = request.POST.get('group_id', '')
  570. user_id = request.POST.get('user_id', '')
  571. photo_id = request.POST.get('photo_id', '')
  572. # 群组用户校验
  573. try:
  574. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
  575. except GroupUserInfo.DoesNotExist:
  576. return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
  577. # 群组照片校验
  578. try:
  579. group_photo = GroupPhotoInfo.objects.get(pk=photo_id)
  580. except GroupPhotoInfo.DoesNotExist:
  581. return response(GroupPhotoStatusCode.GROUP_PHOTO_NOT_FOUND)
  582. # user_id 是否点赞 photo_id
  583. if get_group_photo_thumbup_flag(photo_id, user_id):
  584. return response(GroupPhotoStatusCode.DUPLICATE_THUMB_UP)
  585. # 群组照片点赞记录创建/更新
  586. photo_thumbup, created = PhotoThumbUpInfo.objects.get_or_create(
  587. photo_id=photo_id,
  588. user_id=user_id,
  589. )
  590. photo_thumbup.nickname = group_user.nickname
  591. photo_thumbup.avatar = group_user.avatar
  592. photo_thumbup.thumbup = True
  593. photo_thumbup.save()
  594. # Redis 群组照片点赞数据缓存
  595. set_group_photo_thumbup_flag(photo_id, user_id)
  596. # 群组照片点赞数更新
  597. group_photo.thumbup_num += 1
  598. group_photo.save()
  599. # Redis 群组照片数据缓存
  600. set_group_photo_data(group_id)
  601. # Redis 群组照片点赞列表缓存刷新
  602. set_group_photo_thumbup_list(photo_id)
  603. r.sadd(GROUP_PHOTO_WATCHER_SET % photo_id, user_id)
  604. # 判断群组照片发布者是否已经被管理员移除/主动退出,如若移除/退出,则不给发布者提醒
  605. # 照片所有者(评论/点赞)不给自己提醒
  606. if r.sismember(GROUP_USERS_PASSED_SET % group_photo.group_id, group_photo.user_id) and group_photo.user_id != user_id:
  607. UserMessageInfo.objects.create(
  608. from_uid=user_id,
  609. from_nickname=group_user.nickname,
  610. from_avatar=group_user.avatar,
  611. to_uid=group_photo.user_id,
  612. group_id=group_photo.group_id,
  613. photo_id=group_photo.pk,
  614. msg_type=UserMessageInfo.THUMBUP,
  615. msg_title=u'点赞',
  616. msg_content=u'点赞',
  617. )
  618. # 给所有关注者(评论/点赞)发送提醒
  619. watchers = get_group_photo_watchers(photo_id)
  620. # 从关注者中移除该(评论/点赞)者
  621. watchers.discard(user_id)
  622. # 从关注者中移除该照片所有者
  623. watchers.discard(group_photo.user_id)
  624. for watcher in watchers:
  625. UserMessageInfo.objects.create(
  626. from_uid=user_id,
  627. from_nickname=group_user.nickname,
  628. from_avatar=group_user.avatar,
  629. to_uid=watcher,
  630. group_id=group_photo.group_id,
  631. photo_id=group_photo.pk,
  632. msg_type=UserMessageInfo.THUMBUP,
  633. msg_title=u'点赞',
  634. msg_content=u'点赞',
  635. )
  636. return response(200, 'Thumbup Success', u'点赞提交成功', {
  637. 'thumbup': True,
  638. 'thumbups': get_group_photo_thumbup_list(photo_id),
  639. })
  640. @logit
  641. def thumbup_list_api(request):
  642. """
  643. 飞图点赞列表
  644. :param request:
  645. :return:
  646. """
  647. group_id = request.POST.get('group_id', '')
  648. user_id = request.POST.get('user_id', '')
  649. photo_id = request.POST.get('photo_id', '')
  650. return response(200, 'Get Thumbup List Success', u'获取点赞列表成功', {
  651. 'thumbup': get_group_photo_thumbup_flag(photo_id, user_id), # user_id 是否点赞 photo_id
  652. 'thumbups': get_group_photo_thumbup_list(photo_id), # 群组照片点赞列表
  653. })
  654. @logit
  655. def thumbup_cancel_api(request):
  656. """
  657. 飞图点赞取消
  658. :param request:
  659. :return:
  660. """
  661. group_id = request.POST.get('group_id', '')
  662. user_id = request.POST.get('user_id', '')
  663. photo_id = request.POST.get('photo_id', '')
  664. # 群组用户校验
  665. try:
  666. group_user = GroupUserInfo.objects.get(group_id=group_id, user_id=user_id, user_status=GroupUserInfo.PASSED)
  667. except GroupUserInfo.DoesNotExist:
  668. return response(GroupUserStatusCode.GROUP_USER_NOT_FOUND)
  669. # 群组照片校验
  670. try:
  671. group_photo = GroupPhotoInfo.objects.get(pk=photo_id)
  672. except GroupPhotoInfo.DoesNotExist:
  673. return response(GroupPhotoStatusCode.GROUP_PHOTO_NOT_FOUND)
  674. # user_id 是否点赞 photo_id
  675. if not get_group_photo_thumbup_flag(photo_id, user_id):
  676. return response(GroupPhotoStatusCode.THUMB_UP_NOT_FOUND)
  677. # 群组照片点赞取消
  678. photo_thumbup, created = PhotoThumbUpInfo.objects.get_or_create(
  679. photo_id=photo_id,
  680. user_id=user_id,
  681. )
  682. photo_thumbup.thumbup = False
  683. photo_thumbup.save()
  684. # Redis 群组照片点赞数据移除
  685. del_group_photo_thumbup_flag(photo_id, user_id)
  686. # 群组照片点赞数更新
  687. group_photo.thumbup_num -= 1
  688. group_photo.save()
  689. # Redis 群组照片数据缓存
  690. set_group_photo_data(group_id)
  691. # Redis 群组照片点赞列表缓存刷新
  692. set_group_photo_thumbup_list(photo_id)
  693. # 判断群组照片发布者是否已经被管理员移除/主动退出,如若移除/退出,则不给发布者提醒
  694. if r.sismember(GROUP_USERS_PASSED_SET % group_photo.group_id, group_photo.user_id):
  695. UserMessageInfo.objects.create(
  696. from_uid=user_id,
  697. from_nickname=group_user.nickname,
  698. from_avatar=group_user.avatar,
  699. to_uid=group_photo.user_id,
  700. group_id=group_photo.group_id,
  701. photo_id=group_photo.pk,
  702. msg_type=UserMessageInfo.THUMBUP,
  703. msg_title=u'取消点赞',
  704. msg_content=u'取消点赞',
  705. )
  706. # 群组照片点赞列表
  707. photo_thumbups = PhotoThumbUpInfo.objects.filter(
  708. photo_id=photo_id,
  709. thumbup=True,
  710. )
  711. return response(200, 'Thumbup Cancel Success', u'点赞取消成功', {
  712. 'thumbup': False,
  713. 'thumbups': [thumbup.thumbup_info for thumbup in photo_thumbups],
  714. })
  715. @logit
  716. def pai2_home_api(request):
  717. """
  718. 首页信息
  719. :param request:
  720. :return:
  721. """
  722. user_id = request.POST.get('user_id', '')
  723. page = int(request.POST.get('page', 1))
  724. num = int(request.POST.get('num', settings.PAI2_HOME_PER_PAGE))
  725. # 执行原生 SQL 语句,获取首页照片列表
  726. cursor = connection.cursor()
  727. cursor.execute(PAI2_HOME_API.format(
  728. user_id=user_id,
  729. offset=0,
  730. rows=settings.PAI2_HOME_MAX_ROWS,
  731. ))
  732. rows = cursor.fetchall()
  733. # 使用 records 执行原生 SQL 语句,获取首页照片列表
  734. # rows = db.query(PAI2_HOME_API.format(
  735. # user_id=user_id,
  736. # offset=0,
  737. # rows=settings.PAI2_HOME_MAX_ROWS,
  738. # )).all()
  739. # 首页照片分页
  740. rows, left = pagination(rows, page, num)
  741. # 首页照片信息
  742. rows = [{
  743. 'group_id': row[0],
  744. 'group_name': row[1],
  745. 'group_default_avatar': row[2],
  746. 'group_avatar': row[3],
  747. 'group_from': row[4],
  748. 'photo_id': row[5],
  749. 'photo_url': img_url(row[6]),
  750. 'photo_w': row[7],
  751. 'photo_h': row[8],
  752. 'photo_thumbnail_url': img_url(row[9]),
  753. 'photo_thumbnail_w': row[10],
  754. 'photo_thumbnail_h': row[11],
  755. 'photo_thumbnail2_url': img_url(row[12]),
  756. 'photo_thumbnail2_w': row[13],
  757. 'photo_thumbnail2_h': row[14],
  758. 'photo_share_url': share_url(row[5]), # Warning: Index of This Line is 5
  759. 'user_id': row[15],
  760. 'nickname': row[16],
  761. 'avatar': row[17],
  762. 'comment_num': row[18],
  763. 'thumbup_num': row[19],
  764. 'photo_from': row[20],
  765. 'created_at': row[21],
  766. 'origin_expired_stamps': origin_expired_stamps(row[22], row[15]),
  767. 'thumbup': get_group_photo_thumbup_flag(row[5], user_id),
  768. 'porder': get_lensman_order_record(row[5], user_id) if row[20] == GroupPhotoInfo.SESSION_GROUP else {},
  769. } for row in rows]
  770. # rows = [dict(row) for row in rows]
  771. # [row.update({
  772. # 'photo_url': img_url(row['photo_path']),
  773. # 'photo_thumbnail_url': img_url(row['photo_thumbnail_path']),
  774. # 'photo_thumbnail2_url': img_url(row['photo_thumbnail2_path']),
  775. # 'photo_share_url': share_url(row['photo_id']),
  776. # 'thumbup': get_group_photo_thumbup_flag(row['photo_id'], user_id),
  777. # 'porder': get_lensman_order_record(row['photo_id'], user_id) if row['photo_from'] == GroupPhotoInfo.SESSION_GROUP else {},
  778. # }) for row in rows]
  779. return response(200, 'Get Home Data Success', u'获取首页数据成功', {
  780. 'photos': rows,
  781. 'left': left,
  782. })
  783. @logit
  784. def lensman_photo_price(request):
  785. """
  786. 摄影师照片价格获取
  787. :param request:
  788. :return:
  789. """
  790. user_id = request.POST.get('user_id', '')
  791. photo_id = request.POST.get('photo_id', '')
  792. photo_type = request.POST.get('photo_type', 'nomark') # nomark for 去除水印, origin for 获取高清图
  793. # 处理价格逻辑
  794. lensman_photo_price_key = LENSMAN_PHOTO_PRICE % (user_id, photo_id, photo_type)
  795. lensman_photo_haggle_times_key = LENSMAN_PHOTO_HAGGLE_TIMES % (user_id, photo_id, photo_type)
  796. # Redis 获取存储的价格
  797. price = int(r.get(lensman_photo_price_key) or 0)
  798. if price:
  799. haggle_times = int(r.get(lensman_photo_haggle_times_key) or 0)
  800. # 砍价逻辑
  801. if haggle_times < settings.LENSMAN_PHOTO_HAGGLE_MAX_TIMES:
  802. price -= random.choice([50, 100])
  803. r.incr(lensman_photo_haggle_times_key)
  804. else:
  805. # 获取摄影师定价
  806. # TODO, 此处需要完整的摄影师定价
  807. price = 999 if photo_type == 'origin' else 299
  808. r.set(lensman_photo_price_key, price)
  809. return response(200, 'Get Price Success', u'获取价格成功', {
  810. 'price': price
  811. })
  812. @logit
  813. def lensman_photo_bought(request):
  814. """
  815. 摄影师照片已购买
  816. :param request:
  817. :return:
  818. """
  819. user_id = request.POST.get('user_id', '')
  820. photo_id = request.POST.get('photo_id', '')
  821. return response(200, 'Get Bought Data Success', u'获取购买数据成功', {
  822. 'porder': get_lensman_order_record(photo_id, user_id)
  823. })
  824. @logit
  825. def group_photo_detail(request, photo_id):
  826. photo = GroupPhotoInfo.objects.get(pk=photo_id)
  827. return render(request, 'photo/photo_detail.html', {'photo_url': photo.photo_url, 'api_domain': settings.API_DOMAIN})
  828. @logit
  829. def group_detail(request, group_id):
  830. return render(request, 'page/{}_download.html'.format('ios' if request.iOS else 'adr'), {})
  831. class GroupInfoViewSet(viewsets.ModelViewSet):
  832. queryset = GroupInfo.objects.all().order_by('-pk')
  833. serializer_class = GroupInfoSerializer
  834. class GroupUserInfoViewSet(viewsets.ModelViewSet):
  835. queryset = GroupUserInfo.objects.all().order_by('-pk')
  836. serializer_class = GroupUserInfoSerializer
  837. class GroupPhotoInfoViewSet(viewsets.ModelViewSet):
  838. queryset = GroupPhotoInfo.objects.all().order_by('-pk')
  839. serializer_class = GroupPhotoInfoSerializer
  840. # Only Once Function
  841. def refresh_thumbnail():
  842. """ 刷新缩略图 """
  843. photos = GroupPhotoInfo.objects.filter(status=True)
  844. for photo in photos:
  845. try:
  846. photo_path = photo.photo_path
  847. photo_thumbnail_path = photo_path.replace('.', '_thumbnail.')
  848. photo_thumbnail2_path = photo_path.replace('.', '_thumbnail2.')
  849. # 群组照片缩略图生成
  850. # 双列: 540, 40-50K
  851. photo_w, photo_h, photo_thumbnail_w, photo_thumbnail_h = make_thumbnail(
  852. os.path.join(settings.MEDIA_ROOT, photo_path).replace('\\', '/'),
  853. os.path.join(settings.MEDIA_ROOT, photo_thumbnail_path).replace('\\', '/'),
  854. settings.THUMBNAIL_MAX_WIDTH
  855. )
  856. # 单列: 1080, xx-100K
  857. photo_w, photo_h, photo_thumbnail2_w, photo_thumbnail2_h = make_thumbnail(
  858. os.path.join(settings.MEDIA_ROOT, photo_path).replace('\\', '/'),
  859. os.path.join(settings.MEDIA_ROOT, photo_thumbnail2_path).replace('\\', '/'),
  860. settings.THUMBNAIL_MAX_WIDTH2
  861. )
  862. photo.photo_w = photo_w
  863. photo.photo_h = photo_h
  864. photo.photo_thumbnail_path = photo_thumbnail_path
  865. photo.photo_thumbnail_w = photo_thumbnail_w
  866. photo.photo_thumbnail_h = photo_thumbnail_h
  867. photo.photo_thumbnail2_path = photo_thumbnail2_path
  868. photo.photo_thumbnail2_w = photo_thumbnail2_w
  869. photo.photo_thumbnail2_h = photo_thumbnail2_h
  870. photo.save()
  871. except Exception as e:
  872. pass
  873. return 'Refresh Thumbnail OK'
  874. def statistic_thumbnail_size(pfrom):
  875. """
  876. 统计缩略图大小
  877. :param pfrom: 0 for APP_GROUP, 1 for SESSION_GROUP, -1 for ALL
  878. :return:
  879. """
  880. if pfrom == -1:
  881. photos = GroupPhotoInfo.objects.filter(status=True)
  882. else:
  883. photos = GroupPhotoInfo.objects.filter(photo_from=pfrom, status=True)
  884. photo_count = photos.count()
  885. photo_size = 0
  886. photo_thumbnail_size = 0
  887. photo_thumbnail2_size = 0
  888. for photo in photos:
  889. photo_size += os.path.getsize(os.path.join(settings.MEDIA_ROOT, photo.photo_path).replace('\\', '/'))
  890. photo_thumbnail_size += os.path.getsize(os.path.join(settings.MEDIA_ROOT, photo.photo_thumbnail_path).replace('\\', '/'))
  891. photo_thumbnail2_size += os.path.getsize(os.path.join(settings.MEDIA_ROOT, photo.photo_thumbnail2_path).replace('\\', '/'))
  892. print '>>> Photo Size: %.3f KB' % (photo_size / 1024 / photo_count)
  893. print '>>> Photo Thumbnail Size: %.3f KB' % (photo_thumbnail_size / 1024 / photo_count)
  894. print '>>> Photo Thumbnail2 Size: %.3f KB' % (photo_thumbnail2_size / 1024 / photo_count)
  895. return 'Statistic Thumbnail Size OK'