views.py 64 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718
  1. from datetime import datetime, timedelta
  2. from celery.task.control import revoke
  3. from django.conf import settings
  4. from django.contrib import messages
  5. from django.contrib.auth.decorators import login_required
  6. from django.contrib.postgres.search import SearchQuery
  7. from django.core.mail import EmailMessage
  8. from django.db.models import Q
  9. from django.http import HttpResponseRedirect
  10. from django.shortcuts import get_object_or_404, render
  11. from django.template.defaultfilters import slugify
  12. from drf_yasg import openapi as openapi
  13. from drf_yasg.utils import swagger_auto_schema
  14. from rest_framework import permissions, status
  15. from rest_framework.exceptions import PermissionDenied
  16. from rest_framework.parsers import (
  17. FileUploadParser,
  18. FormParser,
  19. JSONParser,
  20. MultiPartParser,
  21. )
  22. from rest_framework.response import Response
  23. from rest_framework.settings import api_settings
  24. from rest_framework.views import APIView
  25. from actions.models import USER_MEDIA_ACTIONS, MediaAction
  26. from actions.models import USER_VOICE_ACTIONS, VoiceAction
  27. from cms.custom_pagination import FastPaginationWithoutCount
  28. from cms.permissions import IsAuthorizedToAdd, IsUserOrEditor, user_allowed_to_upload
  29. from users.models import User
  30. from .forms import ContactForm, MediaForm, SubtitleForm
  31. from .helpers import clean_query, produce_ffmpeg_commands
  32. from .methods import (
  33. check_comment_for_mention,
  34. get_user_or_session,
  35. is_mediacms_editor,
  36. is_mediacms_manager,
  37. list_tasks,
  38. notify_user_on_comment,
  39. notify_user_on_voice,
  40. show_recommended_media,
  41. show_related_media,
  42. update_user_ratings,
  43. )
  44. from .models import (
  45. Category,
  46. Comment,
  47. EncodeProfile,
  48. Encoding,
  49. Media,
  50. Playlist,
  51. PlaylistMedia,
  52. Tag,
  53. Voice,
  54. )
  55. from .serializers import (
  56. CategorySerializer,
  57. CommentSerializer,
  58. VoiceSerializer,
  59. EncodeProfileSerializer,
  60. MediaSearchSerializer,
  61. MediaSerializer,
  62. PlaylistDetailSerializer,
  63. PlaylistSerializer,
  64. SingleMediaSerializer,
  65. TagSerializer,
  66. )
  67. from .stop_words import STOP_WORDS
  68. from .tasks import save_user_action
  69. from .tasks import save_voice_action
  70. from .tasks import video_with_voices
  71. # TODO: Should we consider USER_VOICE_ACTIONS too?
  72. VALID_USER_ACTIONS = [action for action, name in USER_MEDIA_ACTIONS]
  73. def about(request):
  74. """About view"""
  75. context = {}
  76. return render(request, "cms/about.html", context)
  77. @login_required
  78. def add_subtitle(request):
  79. """Add subtitle view"""
  80. friendly_token = request.GET.get("m", "").strip()
  81. if not friendly_token:
  82. return HttpResponseRedirect("/")
  83. media = Media.objects.filter(friendly_token=friendly_token).first()
  84. if not media:
  85. return HttpResponseRedirect("/")
  86. if not (request.user == media.user or is_mediacms_editor(request.user) or is_mediacms_manager(request.user)):
  87. return HttpResponseRedirect("/")
  88. if request.method == "POST":
  89. form = SubtitleForm(media, request.POST, request.FILES)
  90. if form.is_valid():
  91. subtitle = form.save()
  92. messages.add_message(request, messages.INFO, "Subtitle was added!")
  93. return HttpResponseRedirect(subtitle.media.get_absolute_url())
  94. else:
  95. form = SubtitleForm(media_item=media)
  96. return render(request, "cms/add_subtitle.html", {"form": form})
  97. def categories(request):
  98. """List categories view"""
  99. context = {}
  100. return render(request, "cms/categories.html", context)
  101. def contact(request):
  102. """Contact view"""
  103. context = {}
  104. if request.method == "GET":
  105. form = ContactForm(request.user)
  106. context["form"] = form
  107. else:
  108. form = ContactForm(request.user, request.POST)
  109. if form.is_valid():
  110. if request.user.is_authenticated:
  111. from_email = request.user.email
  112. name = request.user.name
  113. else:
  114. from_email = request.POST.get("from_email")
  115. name = request.POST.get("name")
  116. message = request.POST.get("message")
  117. title = "[{}] - Contact form message received".format(settings.PORTAL_NAME)
  118. msg = """
  119. You have received a message through the contact form\n
  120. Sender name: %s
  121. Sender email: %s\n
  122. \n %s
  123. """ % (
  124. name,
  125. from_email,
  126. message,
  127. )
  128. email = EmailMessage(
  129. title,
  130. msg,
  131. settings.DEFAULT_FROM_EMAIL,
  132. settings.ADMIN_EMAIL_LIST,
  133. reply_to=[from_email],
  134. )
  135. email.send(fail_silently=True)
  136. success_msg = "Message was sent! Thanks for contacting"
  137. context["success_msg"] = success_msg
  138. return render(request, "cms/contact.html", context)
  139. def history(request):
  140. """Show personal history view"""
  141. context = {}
  142. return render(request, "cms/history.html", context)
  143. @login_required
  144. def edit_media(request):
  145. """Edit a media view"""
  146. friendly_token = request.GET.get("m", "").strip()
  147. if not friendly_token:
  148. return HttpResponseRedirect("/")
  149. media = Media.objects.filter(friendly_token=friendly_token).first()
  150. if not media:
  151. return HttpResponseRedirect("/")
  152. if not (request.user == media.user or is_mediacms_editor(request.user) or is_mediacms_manager(request.user)):
  153. return HttpResponseRedirect("/")
  154. if request.method == "POST":
  155. form = MediaForm(request.user, request.POST, request.FILES, instance=media)
  156. if form.is_valid():
  157. media = form.save()
  158. for tag in media.tags.all():
  159. media.tags.remove(tag)
  160. if form.cleaned_data.get("new_tags"):
  161. for tag in form.cleaned_data.get("new_tags").split(","):
  162. tag = slugify(tag)
  163. if tag:
  164. try:
  165. tag = Tag.objects.get(title=tag)
  166. except Tag.DoesNotExist:
  167. tag = Tag.objects.create(title=tag, user=request.user)
  168. if tag not in media.tags.all():
  169. media.tags.add(tag)
  170. messages.add_message(request, messages.INFO, "Media was edited!")
  171. return HttpResponseRedirect(media.get_absolute_url())
  172. else:
  173. form = MediaForm(request.user, instance=media)
  174. return render(
  175. request,
  176. "cms/edit_media.html",
  177. {"form": form, "add_subtitle_url": media.add_subtitle_url},
  178. )
  179. def embed_media(request):
  180. """Embed media view"""
  181. friendly_token = request.GET.get("m", "").strip()
  182. if not friendly_token:
  183. return HttpResponseRedirect("/")
  184. media = Media.objects.values("title").filter(friendly_token=friendly_token).first()
  185. if not media:
  186. return HttpResponseRedirect("/")
  187. context = {}
  188. context["media"] = friendly_token
  189. return render(request, "cms/embed.html", context)
  190. def featured_media(request):
  191. """List featured media view"""
  192. context = {}
  193. return render(request, "cms/featured-media.html", context)
  194. def index(request):
  195. """Index view"""
  196. context = {}
  197. return render(request, "cms/index.html", context)
  198. def latest_media(request):
  199. """List latest media view"""
  200. context = {}
  201. return render(request, "cms/latest-media.html", context)
  202. def liked_media(request):
  203. """List user's liked media view"""
  204. context = {}
  205. return render(request, "cms/liked_media.html", context)
  206. @login_required
  207. def manage_users(request):
  208. """List users management view"""
  209. context = {}
  210. return render(request, "cms/manage_users.html", context)
  211. @login_required
  212. def manage_media(request):
  213. """List media management view"""
  214. context = {}
  215. return render(request, "cms/manage_media.html", context)
  216. @login_required
  217. def manage_comments(request):
  218. """List comments management view"""
  219. context = {}
  220. return render(request, "cms/manage_comments.html", context)
  221. def members(request):
  222. """List members view"""
  223. context = {}
  224. return render(request, "cms/members.html", context)
  225. def recommended_media(request):
  226. """List recommended media view"""
  227. context = {}
  228. return render(request, "cms/recommended-media.html", context)
  229. def search(request):
  230. """Search view"""
  231. context = {}
  232. RSS_URL = f"/rss{request.environ['REQUEST_URI']}"
  233. context["RSS_URL"] = RSS_URL
  234. return render(request, "cms/search.html", context)
  235. def tags(request):
  236. """List tags view"""
  237. context = {}
  238. return render(request, "cms/tags.html", context)
  239. def tos(request):
  240. """Terms of service view"""
  241. context = {}
  242. return render(request, "cms/tos.html", context)
  243. def upload_media(request):
  244. """Upload media view"""
  245. from allauth.account.forms import LoginForm
  246. form = LoginForm()
  247. context = {}
  248. context["form"] = form
  249. context["can_add"] = user_allowed_to_upload(request)
  250. can_upload_exp = settings.CANNOT_ADD_MEDIA_MESSAGE
  251. context["can_upload_exp"] = can_upload_exp
  252. return render(request, "cms/add-media.html", context)
  253. def view_media(request):
  254. """View media view"""
  255. friendly_token = request.GET.get("m", "").strip()
  256. context = {}
  257. media = Media.objects.filter(friendly_token=friendly_token).first()
  258. if not media:
  259. context["media"] = None
  260. return render(request, "cms/media.html", context)
  261. user_or_session = get_user_or_session(request)
  262. save_user_action.delay(user_or_session, friendly_token=friendly_token, action="watch")
  263. context = {}
  264. context["media"] = friendly_token
  265. context["media_object"] = media
  266. context["CAN_DELETE_MEDIA"] = False
  267. context["CAN_EDIT_MEDIA"] = False
  268. context["CAN_DELETE_COMMENTS"] = False
  269. if request.user.is_authenticated:
  270. if (media.user.id == request.user.id) or is_mediacms_editor(request.user) or is_mediacms_manager(request.user):
  271. context["CAN_DELETE_MEDIA"] = True
  272. context["CAN_EDIT_MEDIA"] = True
  273. context["CAN_DELETE_COMMENTS"] = True
  274. return render(request, "cms/media.html", context)
  275. def view_playlist(request, friendly_token):
  276. """View playlist view"""
  277. try:
  278. playlist = Playlist.objects.get(friendly_token=friendly_token)
  279. except BaseException:
  280. playlist = None
  281. context = {}
  282. context["playlist"] = playlist
  283. return render(request, "cms/playlist.html", context)
  284. class MediaList(APIView):
  285. """Media listings views"""
  286. permission_classes = (IsAuthorizedToAdd,)
  287. parser_classes = (MultiPartParser, FormParser, FileUploadParser)
  288. @swagger_auto_schema(
  289. manual_parameters=[
  290. openapi.Parameter(name='page', type=openapi.TYPE_INTEGER, in_=openapi.IN_QUERY, description='Page number'),
  291. openapi.Parameter(name='author', type=openapi.TYPE_STRING, in_=openapi.IN_QUERY, description='username'),
  292. openapi.Parameter(name='show', type=openapi.TYPE_STRING, in_=openapi.IN_QUERY, description='show', enum=['recommended', 'featured', 'latest']),
  293. ],
  294. tags=['Media'],
  295. operation_summary='List Media',
  296. operation_description='Lists all media',
  297. responses={200: MediaSerializer(many=True)},
  298. )
  299. def get(self, request, format=None):
  300. # Show media
  301. params = self.request.query_params
  302. show_param = params.get("show", "")
  303. author_param = params.get("author", "").strip()
  304. if author_param:
  305. user_queryset = User.objects.all()
  306. user = get_object_or_404(user_queryset, username=author_param)
  307. if show_param == "recommended":
  308. pagination_class = FastPaginationWithoutCount
  309. media = show_recommended_media(request, limit=50)
  310. else:
  311. pagination_class = api_settings.DEFAULT_PAGINATION_CLASS
  312. if author_param:
  313. # in case request.user is the user here, show
  314. # all media independant of state
  315. if self.request.user == user:
  316. basic_query = Q(user=user)
  317. else:
  318. basic_query = Q(listable=True, user=user)
  319. else:
  320. # base listings should show safe content
  321. basic_query = Q(listable=True)
  322. if show_param == "featured":
  323. media = Media.objects.filter(basic_query, featured=True)
  324. else:
  325. media = Media.objects.filter(basic_query).order_by("-add_date")
  326. paginator = pagination_class()
  327. if show_param != "recommended":
  328. media = media.prefetch_related("user")
  329. page = paginator.paginate_queryset(media, request)
  330. serializer = MediaSerializer(page, many=True, context={"request": request})
  331. return paginator.get_paginated_response(serializer.data)
  332. @swagger_auto_schema(
  333. manual_parameters=[
  334. openapi.Parameter(name="media_file", in_=openapi.IN_FORM, type=openapi.TYPE_FILE, required=True, description="media_file"),
  335. openapi.Parameter(name="description", in_=openapi.IN_FORM, type=openapi.TYPE_STRING, required=False, description="description"),
  336. openapi.Parameter(name="title", in_=openapi.IN_FORM, type=openapi.TYPE_STRING, required=False, description="title"),
  337. ],
  338. tags=['Media'],
  339. operation_summary='Add new Media',
  340. operation_description='Adds a new media, for authenticated users',
  341. responses={201: openapi.Response('response description', MediaSerializer), 401: 'bad request'},
  342. )
  343. def post(self, request, format=None):
  344. # Add new media
  345. serializer = MediaSerializer(data=request.data, context={"request": request})
  346. if serializer.is_valid():
  347. media_file = request.data["media_file"]
  348. serializer.save(user=request.user, media_file=media_file)
  349. return Response(serializer.data, status=status.HTTP_201_CREATED)
  350. return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
  351. class MediaDetail(APIView):
  352. """
  353. Retrieve, update or delete a media instance.
  354. """
  355. permission_classes = (permissions.IsAuthenticatedOrReadOnly, IsUserOrEditor)
  356. parser_classes = (MultiPartParser, FormParser, FileUploadParser)
  357. def get_object(self, friendly_token, password=None):
  358. try:
  359. media = Media.objects.select_related("user").prefetch_related("encodings__profile").get(friendly_token=friendly_token)
  360. # this need be explicitly called, and will call
  361. # has_object_permission() after has_permission has succeeded
  362. self.check_object_permissions(self.request, media)
  363. if media.state == "private" and not (self.request.user == media.user or is_mediacms_editor(self.request.user)):
  364. if (not password) or (not media.password) or (password != media.password):
  365. return Response(
  366. {"detail": "media is private"},
  367. status=status.HTTP_401_UNAUTHORIZED,
  368. )
  369. return media
  370. except PermissionDenied:
  371. return Response({"detail": "bad permissions"}, status=status.HTTP_401_UNAUTHORIZED)
  372. except BaseException:
  373. return Response(
  374. {"detail": "media file does not exist"},
  375. status=status.HTTP_400_BAD_REQUEST,
  376. )
  377. @swagger_auto_schema(
  378. manual_parameters=[
  379. openapi.Parameter(name='friendly_token', type=openapi.TYPE_STRING, in_=openapi.IN_PATH, description='unique identifier', required=True),
  380. ],
  381. tags=['Media'],
  382. operation_summary='Get information for Media',
  383. operation_description='Get information for a media',
  384. responses={200: SingleMediaSerializer(), 400: 'bad request'},
  385. )
  386. def get(self, request, friendly_token, format=None):
  387. # Get media details
  388. password = request.GET.get("password")
  389. media = self.get_object(friendly_token, password=password)
  390. if isinstance(media, Response):
  391. return media
  392. serializer = SingleMediaSerializer(media, context={"request": request})
  393. if media.state == "private":
  394. related_media = []
  395. else:
  396. related_media = show_related_media(media, request=request, limit=100)
  397. related_media_serializer = MediaSerializer(related_media, many=True, context={"request": request})
  398. related_media = related_media_serializer.data
  399. ret = serializer.data
  400. # update rattings info with user specific ratings
  401. # eg user has already rated for this media
  402. # this only affects user rating and only if enabled
  403. if settings.ALLOW_RATINGS and ret.get("ratings_info") and not request.user.is_anonymous:
  404. ret["ratings_info"] = update_user_ratings(request.user, media, ret.get("ratings_info"))
  405. ret["related_media"] = related_media
  406. return Response(ret)
  407. @swagger_auto_schema(
  408. manual_parameters=[
  409. openapi.Parameter(name='friendly_token', type=openapi.TYPE_STRING, in_=openapi.IN_PATH, description='unique identifier', required=True),
  410. openapi.Parameter(name='type', type=openapi.TYPE_STRING, in_=openapi.IN_FORM, description='action to perform', enum=['encode', 'review']),
  411. openapi.Parameter(
  412. name='encoding_profiles',
  413. type=openapi.TYPE_ARRAY,
  414. items=openapi.Items(type=openapi.TYPE_STRING),
  415. in_=openapi.IN_FORM,
  416. description='if action to perform is encode, need to specify list of ids of encoding profiles',
  417. ),
  418. openapi.Parameter(name='result', type=openapi.TYPE_BOOLEAN, in_=openapi.IN_FORM, description='if action is review, this is the result (True for reviewed, False for not reviewed)'),
  419. ],
  420. tags=['Media'],
  421. operation_summary='Run action on Media',
  422. operation_description='Actions for a media, for MediaCMS editors and managers',
  423. responses={201: 'action created', 400: 'bad request'},
  424. operation_id='media_manager_actions',
  425. )
  426. def post(self, request, friendly_token, format=None):
  427. """superuser actions
  428. Available only to MediaCMS editors and managers
  429. Action is a POST variable, review and encode are implemented
  430. """
  431. media = self.get_object(friendly_token)
  432. if isinstance(media, Response):
  433. return media
  434. if not (is_mediacms_editor(request.user) or is_mediacms_manager(request.user)):
  435. return Response({"detail": "not allowed"}, status=status.HTTP_400_BAD_REQUEST)
  436. action = request.data.get("type")
  437. profiles_list = request.data.get("encoding_profiles")
  438. result = request.data.get("result", True)
  439. if action == "encode":
  440. # Create encoding tasks for specific profiles
  441. valid_profiles = []
  442. if profiles_list:
  443. if isinstance(profiles_list, list):
  444. for p in profiles_list:
  445. p = EncodeProfile.objects.filter(id=p).first()
  446. if p:
  447. valid_profiles.append(p)
  448. elif isinstance(profiles_list, str):
  449. try:
  450. p = EncodeProfile.objects.filter(id=int(profiles_list)).first()
  451. valid_profiles.append(p)
  452. except ValueError:
  453. return Response(
  454. {"detail": "encoding_profiles must be int or list of ints of valid encode profiles"},
  455. status=status.HTTP_400_BAD_REQUEST,
  456. )
  457. media.encode(profiles=valid_profiles)
  458. return Response({"detail": "media will be encoded"}, status=status.HTTP_201_CREATED)
  459. elif action == "review":
  460. if result:
  461. media.is_reviewed = True
  462. elif result is False:
  463. media.is_reviewed = False
  464. media.save(update_fields=["is_reviewed"])
  465. return Response({"detail": "media reviewed set"}, status=status.HTTP_201_CREATED)
  466. return Response(
  467. {"detail": "not valid action or no action specified"},
  468. status=status.HTTP_400_BAD_REQUEST,
  469. )
  470. @swagger_auto_schema(
  471. manual_parameters=[
  472. openapi.Parameter(name="description", in_=openapi.IN_FORM, type=openapi.TYPE_STRING, required=False, description="description"),
  473. openapi.Parameter(name="title", in_=openapi.IN_FORM, type=openapi.TYPE_STRING, required=False, description="title"),
  474. openapi.Parameter(name="media_file", in_=openapi.IN_FORM, type=openapi.TYPE_FILE, required=False, description="media_file"),
  475. ],
  476. tags=['Media'],
  477. operation_summary='Update Media',
  478. operation_description='Update a Media, for Media uploader',
  479. responses={201: openapi.Response('response description', MediaSerializer), 401: 'bad request'},
  480. )
  481. def put(self, request, friendly_token, format=None):
  482. # Update a media object
  483. media = self.get_object(friendly_token)
  484. if isinstance(media, Response):
  485. return media
  486. serializer = MediaSerializer(media, data=request.data, context={"request": request})
  487. if serializer.is_valid():
  488. if request.data.get('media_file'):
  489. media_file = request.data["media_file"]
  490. serializer.save(user=request.user, media_file=media_file)
  491. else:
  492. serializer.save(user=request.user)
  493. return Response(serializer.data, status=status.HTTP_201_CREATED)
  494. return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
  495. @swagger_auto_schema(
  496. manual_parameters=[
  497. openapi.Parameter(name='friendly_token', type=openapi.TYPE_STRING, in_=openapi.IN_PATH, description='unique identifier', required=True),
  498. ],
  499. tags=['Media'],
  500. operation_summary='Delete Media',
  501. operation_description='Delete a Media, for MediaCMS editors and managers',
  502. responses={
  503. 204: 'no content',
  504. },
  505. )
  506. def delete(self, request, friendly_token, format=None):
  507. # Delete a media object
  508. media = self.get_object(friendly_token)
  509. if isinstance(media, Response):
  510. return media
  511. media.delete()
  512. return Response(status=status.HTTP_204_NO_CONTENT)
  513. class MediaActions(APIView):
  514. """
  515. Retrieve, update or delete a media action instance.
  516. """
  517. permission_classes = (permissions.AllowAny,)
  518. parser_classes = (JSONParser,)
  519. def get_object(self, friendly_token):
  520. try:
  521. media = Media.objects.select_related("user").prefetch_related("encodings__profile").get(friendly_token=friendly_token)
  522. if media.state == "private" and self.request.user != media.user:
  523. return Response({"detail": "media is private"}, status=status.HTTP_400_BAD_REQUEST)
  524. return media
  525. except PermissionDenied:
  526. return Response({"detail": "bad permissions"}, status=status.HTTP_400_BAD_REQUEST)
  527. except BaseException:
  528. return Response(
  529. {"detail": "media file does not exist"},
  530. status=status.HTTP_400_BAD_REQUEST,
  531. )
  532. @swagger_auto_schema(
  533. manual_parameters=[],
  534. tags=['Media'],
  535. operation_summary='to_be_written',
  536. operation_description='to_be_written',
  537. )
  538. def get(self, request, friendly_token, format=None):
  539. # show date and reason for each time media was reported
  540. media = self.get_object(friendly_token)
  541. if isinstance(media, Response):
  542. return media
  543. ret = {}
  544. reported = MediaAction.objects.filter(media=media, action="report")
  545. ret["reported"] = []
  546. for rep in reported:
  547. item = {"reported_date": rep.action_date, "reason": rep.extra_info}
  548. ret["reported"].append(item)
  549. return Response(ret, status=status.HTTP_200_OK)
  550. @swagger_auto_schema(
  551. manual_parameters=[],
  552. tags=['Media'],
  553. operation_summary='to_be_written',
  554. operation_description='to_be_written',
  555. )
  556. def post(self, request, friendly_token, format=None):
  557. # perform like/dislike/report actions
  558. media = self.get_object(friendly_token)
  559. if isinstance(media, Response):
  560. return media
  561. action = request.data.get("type")
  562. extra = request.data.get("extra_info")
  563. if request.user.is_anonymous:
  564. # there is a list of allowed actions for
  565. # anonymous users, specified in settings
  566. if action not in settings.ALLOW_ANONYMOUS_ACTIONS:
  567. return Response(
  568. {"detail": "action allowed on logged in users only"},
  569. status=status.HTTP_400_BAD_REQUEST,
  570. )
  571. if action:
  572. user_or_session = get_user_or_session(request)
  573. save_user_action.delay(
  574. user_or_session,
  575. friendly_token=media.friendly_token,
  576. action=action,
  577. extra_info=extra,
  578. )
  579. return Response({"detail": "action received"}, status=status.HTTP_201_CREATED)
  580. else:
  581. return Response({"detail": "no action specified"}, status=status.HTTP_400_BAD_REQUEST)
  582. @swagger_auto_schema(
  583. manual_parameters=[],
  584. tags=['Media'],
  585. operation_summary='to_be_written',
  586. operation_description='to_be_written',
  587. )
  588. def delete(self, request, friendly_token, format=None):
  589. media = self.get_object(friendly_token)
  590. if isinstance(media, Response):
  591. return media
  592. if not request.user.is_superuser:
  593. return Response({"detail": "not allowed"}, status=status.HTTP_400_BAD_REQUEST)
  594. action = request.data.get("type")
  595. if action:
  596. if action == "report": # delete reported actions
  597. MediaAction.objects.filter(media=media, action="report").delete()
  598. media.reported_times = 0
  599. media.save(update_fields=["reported_times"])
  600. return Response(
  601. {"detail": "reset reported times counter"},
  602. status=status.HTTP_201_CREATED,
  603. )
  604. else:
  605. return Response({"detail": "no action specified"}, status=status.HTTP_400_BAD_REQUEST)
  606. class VoiceActions(APIView):
  607. """
  608. Retrieve, update or delete a voice action instance.
  609. """
  610. permission_classes = (permissions.AllowAny,)
  611. parser_classes = (JSONParser,)
  612. def get_object(self, friendly_token):
  613. try:
  614. media = Media.objects.select_related("user").prefetch_related("encodings__profile").get(friendly_token=friendly_token)
  615. if media.state == "private" and self.request.user != media.user:
  616. return Response({"detail": "media is private"}, status=status.HTTP_400_BAD_REQUEST)
  617. return media
  618. except PermissionDenied:
  619. return Response({"detail": "bad permissions"}, status=status.HTTP_400_BAD_REQUEST)
  620. except BaseException:
  621. return Response(
  622. {"detail": "media file does not exist"},
  623. status=status.HTTP_400_BAD_REQUEST,
  624. )
  625. @swagger_auto_schema(
  626. manual_parameters=[],
  627. tags=['Voice Actions'],
  628. operation_summary='to_be_written',
  629. operation_description='to_be_written',
  630. )
  631. def get(self, request, friendly_token, uid=None):
  632. # Show date and info for each time media was liked.
  633. # Test example:
  634. # curl -X GET http://127.0.0.1:80/api/v1/media/dd9TrZxDe/voices/f07f84a8-cf0a-445f-8ae7-568b16ddce55/actions
  635. media = self.get_object(friendly_token)
  636. if isinstance(media, Response):
  637. return media
  638. try:
  639. voice = Voice.objects.get(uid=uid)
  640. except BaseException:
  641. return Response({"detail": "voice does not exist"}, status=status.HTTP_400_BAD_REQUEST,)
  642. ret = {}
  643. like = VoiceAction.objects.filter(voice=voice, action="like")
  644. ret["like"] = []
  645. for lk in like:
  646. item = {"date": lk.action_date, "info": lk.extra_info}
  647. ret["like"].append(item)
  648. likeundo = VoiceAction.objects.filter(voice=voice, action="likeundo")
  649. ret["likeundo"] = []
  650. for lk in likeundo:
  651. item = {"date": lk.action_date, "info": lk.extra_info}
  652. ret["likeundo"].append(item)
  653. return Response(ret, status=status.HTTP_200_OK)
  654. @swagger_auto_schema(
  655. manual_parameters=[],
  656. tags=['Voice Actions'],
  657. operation_summary='to_be_written',
  658. operation_description='to_be_written',
  659. )
  660. def post(self, request, friendly_token, uid=None):
  661. # perform like/dislike/report actions
  662. #
  663. # Test command:
  664. # curl -X POST http://127.0.0.1:80/api/v1/media/dd9TrZxDe/voices/f07f84a8-cf0a-445f-8ae7-568b16ddce55/actions
  665. # Response:
  666. # {"detail":"action allowed on logged in users only"}
  667. media = self.get_object(friendly_token)
  668. if isinstance(media, Response):
  669. return media
  670. # Double-check voice existence.
  671. try:
  672. voice = Voice.objects.get(uid=uid)
  673. except BaseException:
  674. return Response({"detail": "voice does not exist"}, status=status.HTTP_400_BAD_REQUEST,)
  675. action = request.data.get("type")
  676. extra = request.data.get("extra_info")
  677. if request.user.is_anonymous:
  678. # there is a list of allowed actions for
  679. # anonymous users, specified in settings
  680. if action not in settings.ALLOW_ANONYMOUS_ACTIONS:
  681. return Response(
  682. {"detail": "action allowed on logged in users only"},
  683. status=status.HTTP_400_BAD_REQUEST,
  684. )
  685. if action:
  686. user_or_session = get_user_or_session(request)
  687. save_voice_action.delay(
  688. user_or_session,
  689. friendly_token=media.friendly_token,
  690. action=action,
  691. extra_info=extra,
  692. uid=voice.uid,
  693. )
  694. return Response({"detail": "action received"}, status=status.HTTP_201_CREATED)
  695. else:
  696. return Response({"detail": "no action specified"}, status=status.HTTP_400_BAD_REQUEST)
  697. class VideoWithVoices(APIView):
  698. """
  699. Combine a video `media` with some `voices`.
  700. """
  701. permission_classes = (permissions.AllowAny,)
  702. parser_classes = (JSONParser,)
  703. def get_object(self, friendly_token):
  704. try:
  705. media = Media.objects.select_related("user").prefetch_related("encodings__profile").get(friendly_token=friendly_token)
  706. if media.state == "private" and self.request.user != media.user:
  707. return Response({"detail": "media is private"}, status=status.HTTP_400_BAD_REQUEST)
  708. return media
  709. except PermissionDenied:
  710. return Response({"detail": "bad permissions"}, status=status.HTTP_400_BAD_REQUEST)
  711. except BaseException:
  712. return Response(
  713. {"detail": "media file does not exist"},
  714. status=status.HTTP_400_BAD_REQUEST,
  715. )
  716. @swagger_auto_schema(
  717. manual_parameters=[],
  718. tags=['Combine a video media with some voices'],
  719. operation_summary='to_be_written',
  720. operation_description='to_be_written',
  721. )
  722. def post(self, request, friendly_token):
  723. media = self.get_object(friendly_token)
  724. if isinstance(media, Response):
  725. return media
  726. voicesUid = request.data.get("voicesUid")
  727. voicesSrc = request.data.get("voicesSrc")
  728. for uid in voicesUid:
  729. # Double-check voice existence.
  730. try:
  731. voice = Voice.objects.get(uid=uid)
  732. except BaseException:
  733. return Response({"detail": "voice does not exist"}, status=status.HTTP_400_BAD_REQUEST,)
  734. user_or_session = get_user_or_session(request)
  735. video_with_voices.delay(
  736. user_or_session,
  737. friendly_token=media.friendly_token,
  738. voicesUid=voicesUid,
  739. )
  740. return Response({"detail": "video is combined with voices, it's ready for download"}, status=status.HTTP_201_CREATED)
  741. class MediaSearch(APIView):
  742. """
  743. Retrieve results for searc
  744. Only GET is implemented here
  745. """
  746. parser_classes = (JSONParser,)
  747. @swagger_auto_schema(
  748. manual_parameters=[],
  749. tags=['Search'],
  750. operation_summary='to_be_written',
  751. operation_description='to_be_written',
  752. )
  753. def get(self, request, format=None):
  754. params = self.request.query_params
  755. query = params.get("q", "").strip().lower()
  756. category = params.get("c", "").strip()
  757. tag = params.get("t", "").strip()
  758. ordering = params.get("ordering", "").strip()
  759. sort_by = params.get("sort_by", "").strip()
  760. media_type = params.get("media_type", "").strip()
  761. author = params.get("author", "").strip()
  762. upload_date = params.get('upload_date', '').strip()
  763. sort_by_options = ["title", "add_date", "edit_date", "views", "likes"]
  764. if sort_by not in sort_by_options:
  765. sort_by = "add_date"
  766. if ordering == "asc":
  767. ordering = ""
  768. else:
  769. ordering = "-"
  770. if media_type not in ["video", "image", "audio", "pdf"]:
  771. media_type = None
  772. if not (query or category or tag):
  773. ret = {}
  774. return Response(ret, status=status.HTTP_200_OK)
  775. media = Media.objects.filter(state="public", is_reviewed=True)
  776. if query:
  777. # move this processing to a prepare_query function
  778. query = clean_query(query)
  779. q_parts = [q_part.rstrip("y") for q_part in query.split() if q_part not in STOP_WORDS]
  780. if q_parts:
  781. query = SearchQuery(q_parts[0] + ":*", search_type="raw")
  782. for part in q_parts[1:]:
  783. query &= SearchQuery(part + ":*", search_type="raw")
  784. else:
  785. query = None
  786. if query:
  787. media = media.filter(search=query)
  788. if tag:
  789. media = media.filter(tags__title=tag)
  790. if category:
  791. media = media.filter(category__title__contains=category)
  792. if media_type:
  793. media = media.filter(media_type=media_type)
  794. if author:
  795. media = media.filter(user__username=author)
  796. if upload_date:
  797. gte = None
  798. if upload_date == 'today':
  799. gte = datetime.now().date()
  800. if upload_date == 'this_week':
  801. gte = datetime.now() - timedelta(days=7)
  802. if upload_date == 'this_month':
  803. year = datetime.now().date().year
  804. month = datetime.now().date().month
  805. gte = datetime(year, month, 1)
  806. if upload_date == 'this_year':
  807. year = datetime.now().date().year
  808. gte = datetime(year, 1, 1)
  809. if gte:
  810. media = media.filter(add_date__gte=gte)
  811. media = media.order_by(f"{ordering}{sort_by}")
  812. if self.request.query_params.get("show", "").strip() == "titles":
  813. media = media.values("title")[:40]
  814. return Response(media, status=status.HTTP_200_OK)
  815. else:
  816. media = media.prefetch_related("user")
  817. if category or tag:
  818. pagination_class = api_settings.DEFAULT_PAGINATION_CLASS
  819. else:
  820. # pagination_class = FastPaginationWithoutCount
  821. pagination_class = api_settings.DEFAULT_PAGINATION_CLASS
  822. paginator = pagination_class()
  823. page = paginator.paginate_queryset(media, request)
  824. serializer = MediaSearchSerializer(page, many=True, context={"request": request})
  825. return paginator.get_paginated_response(serializer.data)
  826. class PlaylistList(APIView):
  827. """Playlists listings and creation views"""
  828. permission_classes = (permissions.IsAuthenticatedOrReadOnly, IsAuthorizedToAdd)
  829. parser_classes = (JSONParser, MultiPartParser, FormParser, FileUploadParser)
  830. @swagger_auto_schema(
  831. manual_parameters=[],
  832. tags=['Playlists'],
  833. operation_summary='to_be_written',
  834. operation_description='to_be_written',
  835. responses={
  836. 200: openapi.Response('response description', PlaylistSerializer(many=True)),
  837. },
  838. )
  839. def get(self, request, format=None):
  840. pagination_class = api_settings.DEFAULT_PAGINATION_CLASS
  841. paginator = pagination_class()
  842. playlists = Playlist.objects.filter().prefetch_related("user")
  843. if "author" in self.request.query_params:
  844. author = self.request.query_params["author"].strip()
  845. playlists = playlists.filter(user__username=author)
  846. page = paginator.paginate_queryset(playlists, request)
  847. serializer = PlaylistSerializer(page, many=True, context={"request": request})
  848. return paginator.get_paginated_response(serializer.data)
  849. @swagger_auto_schema(
  850. manual_parameters=[],
  851. tags=['Playlists'],
  852. operation_summary='to_be_written',
  853. operation_description='to_be_written',
  854. )
  855. def post(self, request, format=None):
  856. serializer = PlaylistSerializer(data=request.data, context={"request": request})
  857. if serializer.is_valid():
  858. serializer.save(user=request.user)
  859. return Response(serializer.data, status=status.HTTP_201_CREATED)
  860. return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
  861. class PlaylistDetail(APIView):
  862. """Playlist related views"""
  863. permission_classes = (permissions.IsAuthenticatedOrReadOnly, IsUserOrEditor)
  864. parser_classes = (JSONParser, MultiPartParser, FormParser, FileUploadParser)
  865. def get_playlist(self, friendly_token):
  866. try:
  867. playlist = Playlist.objects.get(friendly_token=friendly_token)
  868. self.check_object_permissions(self.request, playlist)
  869. return playlist
  870. except PermissionDenied:
  871. return Response({"detail": "not enough permissions"}, status=status.HTTP_400_BAD_REQUEST)
  872. except BaseException:
  873. return Response(
  874. {"detail": "Playlist does not exist"},
  875. status=status.HTTP_400_BAD_REQUEST,
  876. )
  877. @swagger_auto_schema(
  878. manual_parameters=[],
  879. tags=['Playlists'],
  880. operation_summary='to_be_written',
  881. operation_description='to_be_written',
  882. )
  883. def get(self, request, friendly_token, format=None):
  884. playlist = self.get_playlist(friendly_token)
  885. if isinstance(playlist, Response):
  886. return playlist
  887. serializer = PlaylistDetailSerializer(playlist, context={"request": request})
  888. playlist_media = PlaylistMedia.objects.filter(playlist=playlist).prefetch_related("media__user")
  889. playlist_media = [c.media for c in playlist_media]
  890. playlist_media_serializer = MediaSerializer(playlist_media, many=True, context={"request": request})
  891. ret = serializer.data
  892. ret["playlist_media"] = playlist_media_serializer.data
  893. return Response(ret)
  894. @swagger_auto_schema(
  895. manual_parameters=[],
  896. tags=['Playlists'],
  897. operation_summary='to_be_written',
  898. operation_description='to_be_written',
  899. )
  900. def post(self, request, friendly_token, format=None):
  901. playlist = self.get_playlist(friendly_token)
  902. if isinstance(playlist, Response):
  903. return playlist
  904. serializer = PlaylistDetailSerializer(playlist, data=request.data, context={"request": request})
  905. if serializer.is_valid():
  906. serializer.save(user=request.user)
  907. return Response(serializer.data, status=status.HTTP_201_CREATED)
  908. return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
  909. @swagger_auto_schema(
  910. manual_parameters=[],
  911. tags=['Playlists'],
  912. operation_summary='to_be_written',
  913. operation_description='to_be_written',
  914. )
  915. def put(self, request, friendly_token, format=None):
  916. playlist = self.get_playlist(friendly_token)
  917. if isinstance(playlist, Response):
  918. return playlist
  919. action = request.data.get("type")
  920. media_friendly_token = request.data.get("media_friendly_token")
  921. ordering = 0
  922. if request.data.get("ordering"):
  923. try:
  924. ordering = int(request.data.get("ordering"))
  925. except ValueError:
  926. pass
  927. if action in ["add", "remove", "ordering"]:
  928. media = Media.objects.filter(friendly_token=media_friendly_token).first()
  929. if media:
  930. if action == "add":
  931. media_in_playlist = PlaylistMedia.objects.filter(playlist=playlist).count()
  932. if media_in_playlist >= settings.MAX_MEDIA_PER_PLAYLIST:
  933. return Response(
  934. {"detail": "max number of media for a Playlist reached"},
  935. status=status.HTTP_400_BAD_REQUEST,
  936. )
  937. else:
  938. obj, created = PlaylistMedia.objects.get_or_create(
  939. playlist=playlist,
  940. media=media,
  941. ordering=media_in_playlist + 1,
  942. )
  943. obj.save()
  944. return Response(
  945. {"detail": "media added to Playlist"},
  946. status=status.HTTP_201_CREATED,
  947. )
  948. elif action == "remove":
  949. PlaylistMedia.objects.filter(playlist=playlist, media=media).delete()
  950. return Response(
  951. {"detail": "media removed from Playlist"},
  952. status=status.HTTP_201_CREATED,
  953. )
  954. elif action == "ordering":
  955. if ordering:
  956. playlist.set_ordering(media, ordering)
  957. return Response(
  958. {"detail": "new ordering set"},
  959. status=status.HTTP_201_CREATED,
  960. )
  961. else:
  962. return Response({"detail": "media is not valid"}, status=status.HTTP_400_BAD_REQUEST)
  963. return Response(
  964. {"detail": "invalid or not specified action"},
  965. status=status.HTTP_400_BAD_REQUEST,
  966. )
  967. @swagger_auto_schema(
  968. manual_parameters=[],
  969. tags=['Playlists'],
  970. operation_summary='to_be_written',
  971. operation_description='to_be_written',
  972. )
  973. def delete(self, request, friendly_token, format=None):
  974. playlist = self.get_playlist(friendly_token)
  975. if isinstance(playlist, Response):
  976. return playlist
  977. playlist.delete()
  978. return Response(status=status.HTTP_204_NO_CONTENT)
  979. class EncodingDetail(APIView):
  980. """Experimental. This View is used by remote workers
  981. Needs heavy testing and documentation.
  982. """
  983. permission_classes = (permissions.IsAdminUser,)
  984. parser_classes = (JSONParser, MultiPartParser, FormParser, FileUploadParser)
  985. @swagger_auto_schema(auto_schema=None)
  986. def post(self, request, encoding_id):
  987. ret = {}
  988. force = request.data.get("force", False)
  989. task_id = request.data.get("task_id", False)
  990. action = request.data.get("action", "")
  991. chunk = request.data.get("chunk", False)
  992. chunk_file_path = request.data.get("chunk_file_path", "")
  993. encoding_status = request.data.get("status", "")
  994. progress = request.data.get("progress", "")
  995. commands = request.data.get("commands", "")
  996. logs = request.data.get("logs", "")
  997. retries = request.data.get("retries", "")
  998. worker = request.data.get("worker", "")
  999. temp_file = request.data.get("temp_file", "")
  1000. total_run_time = request.data.get("total_run_time", "")
  1001. if action == "start":
  1002. try:
  1003. encoding = Encoding.objects.get(id=encoding_id)
  1004. media = encoding.media
  1005. profile = encoding.profile
  1006. except BaseException:
  1007. Encoding.objects.filter(id=encoding_id).delete()
  1008. return Response({"status": "fail"}, status=status.HTTP_400_BAD_REQUEST)
  1009. # TODO: break chunk True/False logic here
  1010. if (
  1011. Encoding.objects.filter(
  1012. media=media,
  1013. profile=profile,
  1014. chunk=chunk,
  1015. chunk_file_path=chunk_file_path,
  1016. ).count()
  1017. > 1 # noqa
  1018. and force is False # noqa
  1019. ):
  1020. Encoding.objects.filter(id=encoding_id).delete()
  1021. return Response({"status": "fail"}, status=status.HTTP_400_BAD_REQUEST)
  1022. else:
  1023. Encoding.objects.filter(
  1024. media=media,
  1025. profile=profile,
  1026. chunk=chunk,
  1027. chunk_file_path=chunk_file_path,
  1028. ).exclude(id=encoding.id).delete()
  1029. encoding.status = "running"
  1030. if task_id:
  1031. encoding.task_id = task_id
  1032. encoding.save()
  1033. if chunk:
  1034. original_media_path = chunk_file_path
  1035. original_media_md5sum = encoding.md5sum
  1036. original_media_url = settings.SSL_FRONTEND_HOST + encoding.media_chunk_url
  1037. else:
  1038. original_media_path = media.media_file.path
  1039. original_media_md5sum = media.md5sum
  1040. original_media_url = settings.SSL_FRONTEND_HOST + media.original_media_url
  1041. ret["original_media_url"] = original_media_url
  1042. ret["original_media_path"] = original_media_path
  1043. ret["original_media_md5sum"] = original_media_md5sum
  1044. # generating the commands here, and will replace these with temporary
  1045. # files created on the remote server
  1046. tf = "TEMP_FILE_REPLACE"
  1047. tfpass = "TEMP_FPASS_FILE_REPLACE"
  1048. ffmpeg_commands = produce_ffmpeg_commands(
  1049. original_media_path,
  1050. media.media_info,
  1051. resolution=profile.resolution,
  1052. codec=profile.codec,
  1053. output_filename=tf,
  1054. pass_file=tfpass,
  1055. chunk=chunk,
  1056. )
  1057. if not ffmpeg_commands:
  1058. encoding.delete()
  1059. return Response({"status": "fail"}, status=status.HTTP_400_BAD_REQUEST)
  1060. ret["duration"] = media.duration
  1061. ret["ffmpeg_commands"] = ffmpeg_commands
  1062. ret["profile_extension"] = profile.extension
  1063. return Response(ret, status=status.HTTP_201_CREATED)
  1064. elif action == "update_fields":
  1065. try:
  1066. encoding = Encoding.objects.get(id=encoding_id)
  1067. except BaseException:
  1068. return Response({"status": "fail"}, status=status.HTTP_400_BAD_REQUEST)
  1069. to_update = ["size", "update_date"]
  1070. if encoding_status:
  1071. encoding.status = encoding_status
  1072. to_update.append("status")
  1073. if progress:
  1074. encoding.progress = progress
  1075. to_update.append("progress")
  1076. if logs:
  1077. encoding.logs = logs
  1078. to_update.append("logs")
  1079. if commands:
  1080. encoding.commands = commands
  1081. to_update.append("commands")
  1082. if task_id:
  1083. encoding.task_id = task_id
  1084. to_update.append("task_id")
  1085. if total_run_time:
  1086. encoding.total_run_time = total_run_time
  1087. to_update.append("total_run_time")
  1088. if worker:
  1089. encoding.worker = worker
  1090. to_update.append("worker")
  1091. if temp_file:
  1092. encoding.temp_file = temp_file
  1093. to_update.append("temp_file")
  1094. if retries:
  1095. encoding.retries = retries
  1096. to_update.append("retries")
  1097. try:
  1098. encoding.save(update_fields=to_update)
  1099. except BaseException:
  1100. return Response({"status": "fail"}, status=status.HTTP_400_BAD_REQUEST)
  1101. return Response({"status": "success"}, status=status.HTTP_201_CREATED)
  1102. @swagger_auto_schema(auto_schema=None)
  1103. def put(self, request, encoding_id, format=None):
  1104. encoding_file = request.data["file"]
  1105. encoding = Encoding.objects.filter(id=encoding_id).first()
  1106. if not encoding:
  1107. return Response(
  1108. {"detail": "encoding does not exist"},
  1109. status=status.HTTP_400_BAD_REQUEST,
  1110. )
  1111. encoding.media_file = encoding_file
  1112. encoding.save()
  1113. return Response({"detail": "ok"}, status=status.HTTP_201_CREATED)
  1114. class CommentList(APIView):
  1115. permission_classes = (permissions.IsAuthenticatedOrReadOnly, IsAuthorizedToAdd)
  1116. parser_classes = (JSONParser, MultiPartParser, FormParser, FileUploadParser)
  1117. @swagger_auto_schema(
  1118. manual_parameters=[
  1119. openapi.Parameter(name='page', type=openapi.TYPE_INTEGER, in_=openapi.IN_QUERY, description='Page number'),
  1120. openapi.Parameter(name='author', type=openapi.TYPE_STRING, in_=openapi.IN_QUERY, description='username'),
  1121. ],
  1122. tags=['Comments'],
  1123. operation_summary='Lists Comments',
  1124. operation_description='Paginated listing of all comments',
  1125. responses={
  1126. 200: openapi.Response('response description', CommentSerializer(many=True)),
  1127. },
  1128. )
  1129. def get(self, request, format=None):
  1130. pagination_class = api_settings.DEFAULT_PAGINATION_CLASS
  1131. paginator = pagination_class()
  1132. comments = Comment.objects.filter()
  1133. comments = comments.prefetch_related("user")
  1134. comments = comments.prefetch_related("media")
  1135. params = self.request.query_params
  1136. if "author" in params:
  1137. author_param = params["author"].strip()
  1138. user_queryset = User.objects.all()
  1139. user = get_object_or_404(user_queryset, username=author_param)
  1140. comments = comments.filter(user=user)
  1141. page = paginator.paginate_queryset(comments, request)
  1142. serializer = CommentSerializer(page, many=True, context={"request": request})
  1143. return paginator.get_paginated_response(serializer.data)
  1144. class VoiceList(APIView):
  1145. permission_classes = (permissions.IsAuthenticatedOrReadOnly, IsAuthorizedToAdd)
  1146. parser_classes = (JSONParser, MultiPartParser, FormParser, FileUploadParser)
  1147. @swagger_auto_schema(
  1148. manual_parameters=[
  1149. openapi.Parameter(name='page', type=openapi.TYPE_INTEGER, in_=openapi.IN_QUERY, description='Page number'),
  1150. openapi.Parameter(name='author', type=openapi.TYPE_STRING, in_=openapi.IN_QUERY, description='username'),
  1151. ],
  1152. tags=['Voices'],
  1153. operation_summary='Lists Voices',
  1154. operation_description='Paginated listing of all voices',
  1155. responses={
  1156. 200: openapi.Response('response description', VoiceSerializer(many=True)),
  1157. },
  1158. )
  1159. def get(self, request, format=None):
  1160. pagination_class = api_settings.DEFAULT_PAGINATION_CLASS
  1161. paginator = pagination_class()
  1162. voices = Voice.objects.filter()
  1163. voices = voices.prefetch_related("user")
  1164. voices = voices.prefetch_related("media")
  1165. params = self.request.query_params
  1166. if "author" in params:
  1167. author_param = params["author"].strip()
  1168. user_queryset = User.objects.all()
  1169. user = get_object_or_404(user_queryset, username=author_param)
  1170. voices = voices.filter(user=user)
  1171. page = paginator.paginate_queryset(voices, request)
  1172. serializer = VoiceSerializer(page, many=True, context={"request": request})
  1173. return paginator.get_paginated_response(serializer.data)
  1174. class CommentDetail(APIView):
  1175. """Comments related views
  1176. Listings of comments for a media (GET)
  1177. Create comment (POST)
  1178. Delete comment (DELETE)
  1179. """
  1180. permission_classes = (IsAuthorizedToAdd,)
  1181. parser_classes = (JSONParser, MultiPartParser, FormParser, FileUploadParser)
  1182. def get_object(self, friendly_token):
  1183. try:
  1184. media = Media.objects.select_related("user").get(friendly_token=friendly_token)
  1185. self.check_object_permissions(self.request, media)
  1186. if media.state == "private" and self.request.user != media.user:
  1187. return Response({"detail": "media is private"}, status=status.HTTP_400_BAD_REQUEST)
  1188. return media
  1189. except PermissionDenied:
  1190. return Response({"detail": "bad permissions"}, status=status.HTTP_400_BAD_REQUEST)
  1191. except BaseException:
  1192. return Response(
  1193. {"detail": "media file does not exist"},
  1194. status=status.HTTP_400_BAD_REQUEST,
  1195. )
  1196. @swagger_auto_schema(
  1197. manual_parameters=[],
  1198. tags=['Media'],
  1199. operation_summary='to_be_written',
  1200. operation_description='to_be_written',
  1201. )
  1202. def get(self, request, friendly_token):
  1203. # list comments for a media
  1204. media = self.get_object(friendly_token)
  1205. if isinstance(media, Response):
  1206. return media
  1207. comments = media.comments.filter().prefetch_related("user")
  1208. pagination_class = api_settings.DEFAULT_PAGINATION_CLASS
  1209. paginator = pagination_class()
  1210. page = paginator.paginate_queryset(comments, request)
  1211. serializer = CommentSerializer(page, many=True, context={"request": request})
  1212. return paginator.get_paginated_response(serializer.data)
  1213. @swagger_auto_schema(
  1214. manual_parameters=[],
  1215. tags=['Media'],
  1216. operation_summary='to_be_written',
  1217. operation_description='to_be_written',
  1218. )
  1219. def delete(self, request, friendly_token, uid=None):
  1220. """Delete a comment
  1221. Administrators, MediaCMS editors and managers,
  1222. media owner, and comment owners, can delete a comment
  1223. """
  1224. if uid:
  1225. try:
  1226. comment = Comment.objects.get(uid=uid)
  1227. except BaseException:
  1228. return Response(
  1229. {"detail": "comment does not exist"},
  1230. status=status.HTTP_400_BAD_REQUEST,
  1231. )
  1232. if (comment.user == self.request.user) or comment.media.user == self.request.user or is_mediacms_editor(self.request.user):
  1233. comment.delete()
  1234. else:
  1235. return Response({"detail": "bad permissions"}, status=status.HTTP_400_BAD_REQUEST)
  1236. return Response(status=status.HTTP_204_NO_CONTENT)
  1237. @swagger_auto_schema(
  1238. manual_parameters=[],
  1239. tags=['Media'],
  1240. operation_summary='to_be_written',
  1241. operation_description='to_be_written',
  1242. )
  1243. def post(self, request, friendly_token):
  1244. """Create a comment"""
  1245. media = self.get_object(friendly_token)
  1246. if isinstance(media, Response):
  1247. return media
  1248. if not media.enable_comments:
  1249. return Response(
  1250. {"detail": "comments not allowed here"},
  1251. status=status.HTTP_400_BAD_REQUEST,
  1252. )
  1253. serializer = CommentSerializer(data=request.data, context={"request": request})
  1254. if serializer.is_valid():
  1255. serializer.save(user=request.user, media=media)
  1256. if request.user != media.user:
  1257. notify_user_on_comment(friendly_token=media.friendly_token)
  1258. # here forward the comment to check if a user was mentioned
  1259. if settings.ALLOW_MENTION_IN_COMMENTS:
  1260. check_comment_for_mention(friendly_token=media.friendly_token, comment_text=serializer.data['text'])
  1261. return Response(serializer.data, status=status.HTTP_201_CREATED)
  1262. return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
  1263. class VoiceDetail(APIView):
  1264. """Voices related views
  1265. Listings of voices for a media (GET)
  1266. Create voice (POST)
  1267. Delete voice (DELETE)
  1268. """
  1269. permission_classes = (IsAuthorizedToAdd,)
  1270. parser_classes = (JSONParser, MultiPartParser, FormParser, FileUploadParser)
  1271. def get_object(self, friendly_token):
  1272. try:
  1273. media = Media.objects.select_related("user").get(friendly_token=friendly_token)
  1274. self.check_object_permissions(self.request, media)
  1275. if media.state == "private" and self.request.user != media.user:
  1276. return Response({"detail": "media is private"}, status=status.HTTP_400_BAD_REQUEST)
  1277. return media
  1278. except PermissionDenied:
  1279. return Response({"detail": "bad permissions"}, status=status.HTTP_400_BAD_REQUEST)
  1280. except BaseException:
  1281. return Response(
  1282. {"detail": "media file does not exist"},
  1283. status=status.HTTP_400_BAD_REQUEST,
  1284. )
  1285. @swagger_auto_schema(
  1286. manual_parameters=[],
  1287. tags=['Media'],
  1288. operation_summary='to_be_written',
  1289. operation_description='to_be_written',
  1290. )
  1291. def get(self, request, friendly_token):
  1292. # list voices for a media
  1293. media = self.get_object(friendly_token)
  1294. if isinstance(media, Response):
  1295. return media
  1296. voices = media.voices.filter().prefetch_related("user")
  1297. pagination_class = api_settings.DEFAULT_PAGINATION_CLASS
  1298. paginator = pagination_class()
  1299. page = paginator.paginate_queryset(voices, request)
  1300. serializer = VoiceSerializer(page, many=True, context={"request": request})
  1301. return paginator.get_paginated_response(serializer.data)
  1302. @swagger_auto_schema(
  1303. manual_parameters=[],
  1304. tags=['Media'],
  1305. operation_summary='to_be_written',
  1306. operation_description='to_be_written',
  1307. )
  1308. def delete(self, request, friendly_token, uid=None):
  1309. """Delete a voice
  1310. Administrators, MediaCMS editors and managers,
  1311. media owner, and voice owners, can delete a voice
  1312. """
  1313. if uid:
  1314. # If `uid` is provided,
  1315. # the voice with `uid` would be deleted.
  1316. # Example:
  1317. # DELETE /api/v1/media/A0eMbWrhe/voices/1832ef79-e262-4d79-940c-dd1b717107fb
  1318. try:
  1319. voice = Voice.objects.get(uid=uid)
  1320. except BaseException:
  1321. return Response(
  1322. {"detail": "voice does not exist"},
  1323. status=status.HTTP_400_BAD_REQUEST,
  1324. )
  1325. if (voice.user == self.request.user) or voice.media.user == self.request.user or is_mediacms_editor(self.request.user):
  1326. voice.delete()
  1327. else:
  1328. return Response({"detail": "bad permissions"}, status=status.HTTP_400_BAD_REQUEST)
  1329. else:
  1330. # If `uid` isn't provided,
  1331. # all the voices of `media` created by `self.request.user` would be deleted.
  1332. # Example:
  1333. # DELETE /api/v1/media/A0eMbWrhe/voices
  1334. media = self.get_object(friendly_token)
  1335. if isinstance(media, Response):
  1336. return media
  1337. try:
  1338. voices = media.voices.filter(user=self.request.user).prefetch_related("user")
  1339. except BaseException:
  1340. return Response(
  1341. {"detail:", "BaseException of database query"},
  1342. status=status.HTTP_400_BAD_REQUEST,
  1343. )
  1344. if voices.exists():
  1345. voices.delete()
  1346. else:
  1347. return Response(
  1348. {"detail": "No voice belongs to user"},
  1349. status=status.HTTP_400_BAD_REQUEST,
  1350. )
  1351. return Response(status=status.HTTP_204_NO_CONTENT)
  1352. @swagger_auto_schema(
  1353. manual_parameters=[],
  1354. tags=['Media'],
  1355. operation_summary='to_be_written',
  1356. operation_description='to_be_written',
  1357. )
  1358. def post(self, request, friendly_token):
  1359. """Create a voice"""
  1360. media = self.get_object(friendly_token)
  1361. if isinstance(media, Response):
  1362. return media
  1363. # Currently, recording voices are allowed for all media instances.
  1364. # But they can be forbidden by adding `media.enable_voices` field to DB table model.
  1365. # And then checking it here.
  1366. # Just like `media.enable_comments` field.
  1367. # For now, let's avoid changing `media` DB table model.
  1368. serializer = VoiceSerializer(data=request.data, context={"request": request})
  1369. if serializer.is_valid():
  1370. serializer.save(user=request.user, media=media)
  1371. if request.user != media.user:
  1372. notify_user_on_voice(friendly_token=media.friendly_token)
  1373. return Response(serializer.data, status=status.HTTP_201_CREATED)
  1374. return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
  1375. class UserActions(APIView):
  1376. parser_classes = (JSONParser,)
  1377. @swagger_auto_schema(
  1378. manual_parameters=[
  1379. openapi.Parameter(name='action', type=openapi.TYPE_STRING, in_=openapi.IN_PATH, description='action', required=True, enum=VALID_USER_ACTIONS),
  1380. ],
  1381. tags=['Users'],
  1382. operation_summary='List user actions',
  1383. operation_description='Lists user actions',
  1384. )
  1385. def get(self, request, action):
  1386. media = []
  1387. if action in VALID_USER_ACTIONS:
  1388. if request.user.is_authenticated:
  1389. media = Media.objects.select_related("user").filter(mediaactions__user=request.user, mediaactions__action=action).order_by("-mediaactions__action_date")
  1390. elif request.session.session_key:
  1391. media = (
  1392. Media.objects.select_related("user")
  1393. .filter(
  1394. mediaactions__session_key=request.session.session_key,
  1395. mediaactions__action=action,
  1396. )
  1397. .order_by("-mediaactions__action_date")
  1398. )
  1399. pagination_class = api_settings.DEFAULT_PAGINATION_CLASS
  1400. paginator = pagination_class()
  1401. page = paginator.paginate_queryset(media, request)
  1402. serializer = MediaSerializer(page, many=True, context={"request": request})
  1403. return paginator.get_paginated_response(serializer.data)
  1404. class CategoryList(APIView):
  1405. """List categories"""
  1406. @swagger_auto_schema(
  1407. manual_parameters=[],
  1408. tags=['Categories'],
  1409. operation_summary='Lists Categories',
  1410. operation_description='Lists all categories',
  1411. responses={
  1412. 200: openapi.Response('response description', CategorySerializer),
  1413. },
  1414. )
  1415. def get(self, request, format=None):
  1416. categories = Category.objects.filter().order_by("title")
  1417. serializer = CategorySerializer(categories, many=True, context={"request": request})
  1418. ret = serializer.data
  1419. return Response(ret)
  1420. class TagList(APIView):
  1421. """List tags"""
  1422. @swagger_auto_schema(
  1423. manual_parameters=[
  1424. openapi.Parameter(name='page', type=openapi.TYPE_INTEGER, in_=openapi.IN_QUERY, description='Page number'),
  1425. ],
  1426. tags=['Tags'],
  1427. operation_summary='Lists Tags',
  1428. operation_description='Paginated listing of all tags',
  1429. responses={
  1430. 200: openapi.Response('response description', TagSerializer),
  1431. },
  1432. )
  1433. def get(self, request, format=None):
  1434. tags = Tag.objects.filter().order_by("-media_count")
  1435. pagination_class = api_settings.DEFAULT_PAGINATION_CLASS
  1436. paginator = pagination_class()
  1437. page = paginator.paginate_queryset(tags, request)
  1438. serializer = TagSerializer(page, many=True, context={"request": request})
  1439. return paginator.get_paginated_response(serializer.data)
  1440. class EncodeProfileList(APIView):
  1441. """List encode profiles"""
  1442. @swagger_auto_schema(
  1443. manual_parameters=[],
  1444. tags=['Encoding Profiles'],
  1445. operation_summary='List Encoding Profiles',
  1446. operation_description='Lists all encoding profiles for videos',
  1447. responses={200: EncodeProfileSerializer(many=True)},
  1448. )
  1449. def get(self, request, format=None):
  1450. profiles = EncodeProfile.objects.all()
  1451. serializer = EncodeProfileSerializer(profiles, many=True, context={"request": request})
  1452. return Response(serializer.data)
  1453. class TasksList(APIView):
  1454. """List tasks"""
  1455. swagger_schema = None
  1456. permission_classes = (permissions.IsAdminUser,)
  1457. def get(self, request, format=None):
  1458. ret = list_tasks()
  1459. return Response(ret)
  1460. class TaskDetail(APIView):
  1461. """Cancel a task"""
  1462. swagger_schema = None
  1463. permission_classes = (permissions.IsAdminUser,)
  1464. def delete(self, request, uid, format=None):
  1465. revoke(uid, terminate=True)
  1466. return Response(status=status.HTTP_204_NO_CONTENT)