拍爱

views.py 35KB

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