views.py 45 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076
  1. import datetime
  2. import random
  3. import bleach
  4. import pytz
  5. from django.db.models import Q
  6. from django.http import HttpResponse
  7. from django.shortcuts import render, redirect, get_object_or_404
  8. from django.utils.html import strip_tags
  9. import apps
  10. from apps.main.models import Playlist, Tag
  11. from django.contrib.auth.decorators import login_required # redirects user to settings.LOGIN_URL
  12. from allauth.socialaccount.models import SocialToken
  13. from django.views.decorators.http import require_POST
  14. from django.contrib import messages
  15. from django.template import loader
  16. from .util import *
  17. # Create your views here.
  18. @login_required
  19. def home(request):
  20. user_profile = request.user
  21. watching = user_profile.playlists.filter(Q(marked_as="watching") & Q(is_in_db=True)).order_by("-num_of_accesses")
  22. recently_accessed_playlists = user_profile.playlists.filter(is_in_db=True).filter(
  23. updated_at__gt=user_profile.profile.updated_at).order_by("-updated_at")[:6]
  24. recently_added_playlists = user_profile.playlists.filter(is_in_db=True).order_by("-created_at")[:6]
  25. #### FOR NEWLY JOINED USERS ######
  26. channel_found = True
  27. if user_profile.profile.show_import_page:
  28. """
  29. Logic:
  30. show_import_page is True by default. When a user logs in for the first time (infact anytime), google
  31. redirects them to 'home' url. Since, show_import_page is True by default, the user is then redirected
  32. from 'home' to 'import_in_progress' url
  33. show_import_page is only set false in the import_in_progress.html page, i.e when user cancels YT import
  34. """
  35. # user_profile.show_import_page = False
  36. if user_profile.profile.access_token.strip() == "" or user_profile.profile.refresh_token.strip() == "":
  37. user_social_token = SocialToken.objects.get(account__user=request.user)
  38. user_profile.profile.access_token = user_social_token.token
  39. user_profile.profile.refresh_token = user_social_token.token_secret
  40. user_profile.profile.expires_at = user_social_token.expires_at
  41. user_profile.save()
  42. Playlist.objects.getUserYTChannelID(user_profile)
  43. if user_profile.profile.imported_yt_playlists:
  44. user_profile.profile.show_import_page = False # after user imports all their YT playlists no need to show_import_page again
  45. user_profile.profile.save(update_fields=['show_import_page'])
  46. return render(request, "home.html", {"import_successful": True})
  47. return render(request, "import_in_progress.html")
  48. # if Playlist.objects.getUserYTChannelID(request.user) == -1: # user channel not found
  49. # channel_found = False
  50. # else:
  51. # Playlist.objects.initPlaylist(request.user, None) # get all playlists from user's YT channel
  52. # return render(request, "home.html", {"import_successful": True})
  53. ##################################
  54. user_playlists = request.user.playlists.filter(is_in_db=True)
  55. total_num_playlists = user_playlists.count()
  56. user_playlists = user_playlists.filter(num_of_accesses__gt=0).order_by(
  57. "-num_of_accesses")
  58. statistics = {
  59. "public_x": 0,
  60. "private_x": 0,
  61. "favorites_x": 0,
  62. "watching_x": 0,
  63. "imported_x": 0
  64. }
  65. if total_num_playlists != 0:
  66. # x means percentage
  67. statistics["public_x"] = round(user_playlists.filter(is_private_on_yt=False).count() / total_num_playlists, 1) * 100
  68. statistics["private_x"] = round(user_playlists.filter(is_private_on_yt=True).count() / total_num_playlists, 1) * 100
  69. statistics["favorites_x"] = round(user_playlists.filter(is_favorite=True).count() / total_num_playlists, 1) * 100
  70. statistics["watching_x"] = round(user_playlists.filter(marked_as="watching").count() / total_num_playlists, 1) * 100
  71. statistics["imported_x"] = round(user_playlists.filter(is_user_owned=False).count() / total_num_playlists, 1) * 100
  72. return render(request, 'home.html', {"channel_found": channel_found,
  73. "user_playlists": user_playlists,
  74. "watching": watching,
  75. "recently_accessed_playlists": recently_accessed_playlists,
  76. "recently_added_playlists": recently_added_playlists,
  77. "statistics": statistics})
  78. @login_required
  79. def view_video(request, video_id):
  80. if request.user.videos.filter(video_id=video_id).exists():
  81. video = request.user.videos.get(video_id=video_id)
  82. if video.is_unavailable_on_yt or video.was_deleted_on_yt:
  83. messages.error(request, "Video went private/deleted on YouTube!")
  84. return redirect('home')
  85. return render(request, 'view_video.html', {"video": video})
  86. else:
  87. messages.error(request, "No such video in your UnTube collection!")
  88. return redirect('home')
  89. @login_required
  90. @require_POST
  91. def video_notes(request, video_id):
  92. print(request.POST)
  93. if request.user.videos.filter(video_id=video_id).exists():
  94. video = request.user.videos.get(video_id=video_id)
  95. if 'video-notes-text-area' in request.POST:
  96. video.user_notes = bleach.clean(request.POST['video-notes-text-area'], tags=['br'])
  97. video.save(update_fields=['user_notes', 'user_label'])
  98. # messages.success(request, 'Saved!')
  99. return HttpResponse("""
  100. <div hx-ext="class-tools">
  101. <div classes="add visually-hidden:2s">Saved!</div>
  102. </div>
  103. """)
  104. else:
  105. return HttpResponse('No such video in your UnTube collection!')
  106. @login_required
  107. def view_playlist(request, playlist_id):
  108. user_profile = request.user
  109. user_owned_playlists = user_profile.playlists.filter(Q(is_user_owned=True) & Q(is_in_db=True))
  110. # specific playlist requested
  111. if user_profile.playlists.filter(Q(playlist_id=playlist_id) & Q(is_in_db=True)).exists():
  112. playlist = user_profile.playlists.get(playlist_id__exact=playlist_id)
  113. # playlist.num_of_accesses += 1
  114. # only note down that the playlist as been viewed when 5mins has passed since the last access
  115. if playlist.last_accessed_on + datetime.timedelta(minutes=5) < datetime.datetime.now(pytz.utc):
  116. playlist.num_of_accesses += 1
  117. playlist.last_accessed_on = datetime.datetime.now(pytz.utc)
  118. playlist.save()
  119. else:
  120. messages.error(request, "No such playlist found!")
  121. return redirect('home')
  122. if playlist.has_new_updates:
  123. recently_updated_videos = playlist.videos.filter(video_details_modified=True)
  124. for video in recently_updated_videos:
  125. if video.video_details_modified_at + datetime.timedelta(hours=12) < datetime.datetime.now(
  126. pytz.utc): # expired
  127. video.video_details_modified = False
  128. video.save()
  129. if not recently_updated_videos.exists():
  130. playlist.has_new_updates = False
  131. playlist.save()
  132. playlist_items = playlist.playlist_items.select_related('video').order_by("video_position")
  133. user_created_tags = Tag.objects.filter(created_by=request.user)
  134. playlist_tags = playlist.tags.all()
  135. for tag in playlist_tags:
  136. tag.times_viewed += 1
  137. tag.save(update_fields=['times_viewed'])
  138. unused_tags = user_created_tags.difference(playlist_tags)
  139. return render(request, 'view_playlist.html', {"playlist": playlist,
  140. "playlist_tags": playlist_tags,
  141. "unused_tags": unused_tags,
  142. "playlist_items": playlist_items,
  143. "user_owned_playlists": user_owned_playlists,
  144. "watching_message": generateWatchingMessage(playlist),
  145. })
  146. @login_required
  147. def tagged_playlists(request, tag):
  148. tag = get_object_or_404(Tag, created_by=request.user, name=tag)
  149. playlists = tag.playlists.all()
  150. return render(request, 'all_playlists_with_tag.html', {"playlists": playlists, "tag": tag})
  151. @login_required
  152. def all_playlists(request, playlist_type):
  153. """
  154. Possible playlist types for marked_as attribute: (saved in database like this)
  155. "none", "watching", "plan-to-watch"
  156. """
  157. playlist_type = playlist_type.lower()
  158. watching = False
  159. if playlist_type == "" or playlist_type == "all":
  160. playlists = request.user.playlists.all().filter(is_in_db=True)
  161. playlist_type_display = "All Playlists"
  162. elif playlist_type == "user-owned": # YT playlists owned by user
  163. playlists = request.user.playlists.all().filter(Q(is_user_owned=True) & Q(is_in_db=True))
  164. playlist_type_display = "Your YouTube Playlists"
  165. elif playlist_type == "imported": # YT playlists (public) owned by others
  166. playlists = request.user.playlists.all().filter(Q(is_user_owned=False) & Q(is_in_db=True))
  167. playlist_type_display = "Imported playlists"
  168. elif playlist_type == "favorites": # YT playlists (public) owned by others
  169. playlists = request.user.playlists.all().filter(Q(is_favorite=True) & Q(is_in_db=True))
  170. playlist_type_display = "Favorites"
  171. elif playlist_type.lower() in ["watching", "plan-to-watch"]:
  172. playlists = request.user.playlists.filter(Q(marked_as=playlist_type.lower()) & Q(is_in_db=True))
  173. playlist_type_display = playlist_type.lower().replace("-", " ")
  174. if playlist_type.lower() == "watching":
  175. watching = True
  176. elif playlist_type.lower() == "home": # displays cards of all playlist types
  177. return render(request, 'playlists_home.html')
  178. elif playlist_type.lower() == "random": # randomize playlist
  179. if request.method == "POST":
  180. playlists_type = request.POST["playlistsType"]
  181. if playlists_type == "All":
  182. playlists = request.user.playlists.all().filter(is_in_db=True)
  183. elif playlists_type == "Favorites":
  184. playlists = request.user.playlists.all().filter(Q(is_favorite=True) & Q(is_in_db=True))
  185. elif playlists_type == "Watching":
  186. playlists = request.user.playlists.filter(Q(marked_as="watching") & Q(is_in_db=True))
  187. elif playlists_type == "Plan to Watch":
  188. playlists = request.user.playlists.filter(Q(marked_as="plan-to-watch") & Q(is_in_db=True))
  189. else:
  190. return redirect('/playlists/home')
  191. if not playlists.exists():
  192. messages.info(request, f"No playlists in {playlists_type}")
  193. return redirect('/playlists/home')
  194. random_playlist = random.choice(playlists)
  195. return redirect(f'/playlist/{random_playlist.playlist_id}')
  196. return render(request, 'playlists_home.html')
  197. else:
  198. return redirect('home')
  199. return render(request, 'all_playlists.html', {"playlists": playlists,
  200. "playlist_type": playlist_type,
  201. "playlist_type_display": playlist_type_display,
  202. "watching": watching})
  203. @login_required
  204. def all_videos(request, videos_type):
  205. """
  206. To implement this need to redesign the database
  207. Currently videos -> playlist -> user.profile
  208. Need to do
  209. user.profile <- videos <- playlistItem -> playlist
  210. many ways actually
  211. """
  212. videos_type = videos_type.lower()
  213. if videos_type == "" or videos_type == "all":
  214. playlists = request.user.playlists.all().filter(is_in_db=True)
  215. videos_type_display = "All Videos"
  216. elif videos_type == "user-owned": # YT playlists owned by user
  217. playlists = request.user.playlists.all().filter(Q(is_user_owned=True) & Q(is_in_db=True))
  218. videos_type_display = "All Videos in your YouTube Playlists"
  219. elif videos_type == "imported": # YT playlists (public) owned by others
  220. playlists = request.user.playlists.all().filter(Q(is_user_owned=False) & Q(is_in_db=True))
  221. videos_type_display = "Imported YouTube Playlists Videos"
  222. elif videos_type == "favorites": # YT playlists (public) owned by others
  223. playlists = request.user.playlists.all().filter(Q(is_favorite=True) & Q(is_in_db=True))
  224. videos_type_display = "Favorite Videos"
  225. elif videos_type == "watched": # YT playlists (public) owned by others
  226. playlists = request.user.playlists.all().filter(Q(is_favorite=True) & Q(is_in_db=True))
  227. videos_type_display = "Watched Videos"
  228. elif videos_type == 'hidden-videos': # YT playlists (public) owned by others
  229. playlists = request.user.playlists.all().filter(Q(is_favorite=True) & Q(is_in_db=True))
  230. videos_type_display = "Hidden Videos"
  231. elif videos_type.lower() == "home": # displays cards of all playlist types
  232. return render(request, 'videos_home.html')
  233. else:
  234. return redirect('home')
  235. return render(request, 'all_playlists.html', {"playlists": playlists,
  236. "videos_type": videos_type,
  237. "videos_type_display": videos_type_display})
  238. @login_required
  239. def order_playlist_by(request, playlist_id, order_by):
  240. playlist = request.user.playlists.get(Q(playlist_id=playlist_id) & Q(is_in_db=True))
  241. display_text = "Nothing in this playlist! Add something!" # what to display when requested order/filter has no videws
  242. videos_details = ""
  243. if order_by == "all":
  244. playlist_items = playlist.playlist_items.select_related('video').order_by("video_position")
  245. elif order_by == "favorites":
  246. playlist_items = playlist.playlist_items.select_related('video').filter(video__is_favorite=True).order_by("video_position")
  247. videos_details = "Sorted by Favorites"
  248. display_text = "No favorites yet!"
  249. elif order_by == "popularity":
  250. videos_details = "Sorted by Popularity"
  251. videos = playlist.videos.order_by("-like_count")
  252. elif order_by == "date-published":
  253. videos_details = "Sorted by Date Published"
  254. videos = playlist.videos.order_by("-published_at")
  255. elif order_by == "views":
  256. videos_details = "Sorted by View Count"
  257. videos = playlist.videos.order_by("-view_count")
  258. elif order_by == "has-cc":
  259. videos_details = "Filtered by Has CC"
  260. videos = playlist.videos.filter(has_cc=True).order_by("video_position")
  261. display_text = "No videos in this playlist have CC :("
  262. elif order_by == "duration":
  263. videos_details = "Sorted by Video Duration"
  264. videos = playlist.videos.order_by("-duration_in_seconds")
  265. elif order_by == 'new-updates':
  266. playlist_items = []
  267. videos_details = "Sorted by New Updates"
  268. display_text = "No new updates! Note that deleted videos will not show up here."
  269. if playlist.has_new_updates:
  270. recently_updated_videos = playlist.playlist_items.select_related('video').filter(video__video_details_modified=True)
  271. for playlist_item in recently_updated_videos:
  272. if playlist_item.video.video_details_modified_at + datetime.timedelta(hours=12) < datetime.datetime.now(
  273. pytz.utc): # expired
  274. playlist_item.video.video_details_modified = False
  275. playlist_item.video.save(update_fields=['video_details_modified'])
  276. if not recently_updated_videos.exists():
  277. playlist.has_new_updates = False
  278. playlist.save(update_fields=['has_new_updates'])
  279. else:
  280. playlist_items = recently_updated_videos.order_by("video_position")
  281. elif order_by == 'unavailable-videos':
  282. videos = playlist.videos.filter(Q(is_unavailable_on_yt=True) & Q(was_deleted_on_yt=True))
  283. videos_details = "Sorted by Unavailable Videos"
  284. display_text = "None of the videos in this playlist have gone unavailable... yet."
  285. else:
  286. return redirect('home')
  287. return HttpResponse(loader.get_template("intercooler/videos.html").render({"playlist": playlist,
  288. "playlist_items": playlist_items,
  289. "videos_details": videos_details,
  290. "display_text": display_text,
  291. "order_by": order_by}))
  292. @login_required
  293. def order_playlists_by(request, playlist_type, order_by):
  294. print("GET", request.GET)
  295. print("POST", request.POST)
  296. print("CONTENT PARAMS", request.content_params)
  297. print("HEAD", request.headers)
  298. print("BODY", request.body)
  299. watching = False
  300. if playlist_type == "" or playlist_type.lower() == "all":
  301. playlists = request.user.playlists.all()
  302. elif playlist_type.lower() == "favorites":
  303. playlists = request.user.playlists.filter(Q(is_favorite=True) & Q(is_in_db=True))
  304. elif playlist_type.lower() in ["watching", "plan-to-watch"]:
  305. playlists = request.user.playlists.filter(Q(marked_as=playlist_type.lower()) & Q(is_in_db=True))
  306. if playlist_type.lower() == "watching":
  307. watching = True
  308. elif playlist_type.lower() == "imported":
  309. playlists = request.user.playlists.filter(Q(is_user_owned=False) & Q(is_in_db=True))
  310. elif playlist_type.lower() == "user-owned":
  311. playlists = request.user.playlists.filter(Q(is_user_owned=True) & Q(is_in_db=True))
  312. else:
  313. return HttpResponse("Not found.")
  314. if order_by == 'recently-accessed':
  315. playlists = playlists.order_by("-updated_at")
  316. elif order_by == 'playlist-duration-in-seconds':
  317. playlists = playlists.order_by("-playlist_duration_in_seconds")
  318. elif order_by == 'video-count':
  319. playlists = playlists.order_by("-video_count")
  320. return HttpResponse(loader.get_template("intercooler/playlists.html")
  321. .render({"playlists": playlists, "watching": watching}))
  322. @login_required
  323. def mark_playlist_as(request, playlist_id, mark_as):
  324. playlist = request.user.playlists.get(playlist_id=playlist_id)
  325. marked_as_response = '<span></span><meta http-equiv="refresh" content="0" />'
  326. if mark_as in ["watching", "on-hold", "plan-to-watch"]:
  327. playlist.marked_as = mark_as
  328. playlist.save()
  329. icon = ""
  330. if mark_as == "watching":
  331. playlist.last_watched = datetime.datetime.now(pytz.utc)
  332. playlist.save(update_fields=['last_watched'])
  333. icon = '<i class="fas fa-fire-alt me-2"></i>'
  334. elif mark_as == "plan-to-watch":
  335. icon = '<i class="fas fa-flag me-2"></i>'
  336. marked_as_response = f'<span class="badge bg-success text-white" >{icon}{mark_as}</span> <meta http-equiv="refresh" content="0" />'
  337. elif mark_as == "none":
  338. playlist.marked_as = mark_as
  339. playlist.save()
  340. elif mark_as == "favorite":
  341. if playlist.is_favorite:
  342. playlist.is_favorite = False
  343. playlist.save()
  344. return HttpResponse('<i class="far fa-star"></i>')
  345. else:
  346. playlist.is_favorite = True
  347. playlist.save()
  348. return HttpResponse('<i class="fas fa-star"></i>')
  349. else:
  350. return redirect('home')
  351. return HttpResponse(marked_as_response)
  352. @login_required
  353. def playlists_home(request):
  354. return render(request, 'playlists_home.html')
  355. @login_required
  356. @require_POST
  357. def delete_videos(request, playlist_id, command):
  358. video_ids = request.POST.getlist("video-id", default=[])
  359. print(request.POST)
  360. num_vids = len(video_ids)
  361. extra_text = " "
  362. if num_vids == 0:
  363. return HttpResponse("<h5>Select some videos first!</h5><hr>")
  364. if 'confirm before deleting' in request.POST:
  365. if request.POST['confirm before deleting'] == 'False':
  366. command = "confirmed"
  367. if command == "confirm":
  368. print(video_ids)
  369. if num_vids == request.user.playlists.get(playlist_id=playlist_id).videos.all().count():
  370. delete_text = "ALL VIDEOS"
  371. extra_text = " This will not delete the playlist itself, will only make the playlist empty. "
  372. else:
  373. delete_text = f"{num_vids} videos"
  374. return HttpResponse(
  375. f"""<h5>
  376. Are you sure you want to delete {delete_text} from your YouTube playlist?{extra_text}This cannot be undone.</h5>
  377. <button hx-post="/from/{playlist_id}/delete-videos/confirmed" hx-include="[id='video-checkboxes']" hx-target="#delete-videos-confirm-box" type="button" class="btn btn-outline-danger btn-sm">Confirm</button>
  378. <hr>
  379. """)
  380. elif command == "confirmed":
  381. print(video_ids)
  382. url = f"/from/{playlist_id}/delete-videos/start"
  383. return HttpResponse(
  384. f"""
  385. <div class="spinner-border text-light" role="status" hx-post="{url}" hx-trigger="load" hx-include="[id='video-checkboxes']" hx-target="#delete-videos-confirm-box"></div><hr>
  386. """)
  387. elif command == "start":
  388. print("Deleting", len(video_ids), "videos")
  389. Playlist.objects.deletePlaylistItems(request.user, playlist_id, video_ids)
  390. # playlist = request.user.playlists.get(playlist_id=playlist_id)
  391. # playlist.has_playlist_changed = True
  392. # playlist.save(update_fields=['has_playlist_changed'])
  393. return HttpResponse(f"""
  394. <h5 hx-get="/playlist/{playlist_id}/update/checkforupdates" hx-trigger="load delay:3s" hx-target="#checkforupdates">
  395. Done deleting selected videos from your playlist on YouTube. Playlist on UnTube will update soon.
  396. </h5>
  397. <hr>
  398. """)
  399. @login_required
  400. @require_POST
  401. def search_tagged_playlists(request, tag):
  402. tag = get_object_or_404(Tag, created_by=request.user, name=tag)
  403. playlists = tag.playlists.all()
  404. return HttpResponse("yay")
  405. @login_required
  406. @require_POST
  407. def search_playlists(request, playlist_type):
  408. # print(request.POST) # prints <QueryDict: {'search': ['aa']}>
  409. search_query = request.POST["search"]
  410. watching = False
  411. playlists = None
  412. if playlist_type == "all":
  413. try:
  414. playlists = request.user.playlists.all().filter(Q(name__startswith=search_query) & Q(is_in_db=True))
  415. except:
  416. playlists = request.user.playlists.all()
  417. elif playlist_type == "user-owned": # YT playlists owned by user
  418. try:
  419. playlists = request.user.playlists.filter(
  420. Q(name__startswith=search_query) & Q(is_user_owned=True) & Q(is_in_db=True))
  421. except:
  422. playlists = request.user.playlists.filter(Q(is_user_owned=True) & Q(is_in_db=True))
  423. elif playlist_type == "imported": # YT playlists (public) owned by others
  424. try:
  425. playlists = request.user.playlists.filter(
  426. Q(name__startswith=search_query) & Q(is_user_owned=False) & Q(is_in_db=True))
  427. except:
  428. playlists = request.user.playlists.filter(Q(is_user_owned=True) & Q(is_in_db=True))
  429. elif playlist_type == "favorites": # YT playlists (public) owned by others
  430. try:
  431. playlists = request.user.playlists.filter(
  432. Q(name__startswith=search_query) & Q(is_favorite=True) & Q(is_in_db=True))
  433. except:
  434. playlists = request.user.playlists.filter(Q(is_favorite=True) & Q(is_in_db=True))
  435. elif playlist_type in ["watching", "plan-to-watch"]:
  436. try:
  437. playlists = request.user.playlists.filter(
  438. Q(name__startswith=search_query) & Q(marked_as=playlist_type) & Q(is_in_db=True))
  439. except:
  440. playlists = request.user.playlists.all().filter(Q(marked_as=playlist_type) & Q(is_in_db=True))
  441. if playlist_type == "watching":
  442. watching = True
  443. return HttpResponse(loader.get_template("intercooler/playlists.html")
  444. .render({"playlists": playlists,
  445. "watching": watching}))
  446. #### MANAGE VIDEOS #####
  447. @login_required
  448. def mark_video_favortie(request, playlist_id, video_id):
  449. video = request.user.videos.get(video_id=video_id)
  450. if video.is_favorite:
  451. video.is_favorite = False
  452. video.save(update_fields=['is_favorite'])
  453. return HttpResponse('<i class="far fa-heart"></i>')
  454. else:
  455. video.is_favorite = True
  456. video.save(update_fields=['is_favorite'])
  457. return HttpResponse('<i class="fas fa-heart" style="color: #fafa06"></i>')
  458. @login_required
  459. def mark_video_watched(request, playlist_id, video_id):
  460. playlist = request.user.playlists.get(playlist_id=playlist_id)
  461. video = playlist.videos.get(video_id=video_id)
  462. if video.is_marked_as_watched:
  463. video.is_marked_as_watched = False
  464. video.save(update_fields=['is_marked_as_watched'])
  465. return HttpResponse(
  466. f'<i class="far fa-check-circle" hx-get="/playlist/{playlist_id}/get-watch-message" hx-trigger="load" hx-target="#playlist-watch-message"></i>')
  467. else:
  468. video.is_marked_as_watched = True
  469. video.save(update_fields=['is_marked_as_watched'])
  470. playlist.last_watched = datetime.datetime.now(pytz.utc)
  471. playlist.save(update_fields=['last_watched'])
  472. return HttpResponse(
  473. f'<i class="fas fa-check-circle" hx-get="/playlist/{playlist_id}/get-watch-message" hx-trigger="load" hx-target="#playlist-watch-message"></i>')
  474. ###########
  475. @login_required
  476. def search(request):
  477. if request.method == "GET":
  478. return render(request, 'search_untube_page.html', {"playlists": request.user.playlists.all()})
  479. else:
  480. return redirect('home')
  481. @login_required
  482. @require_POST
  483. def search_UnTube(request):
  484. print(request.POST)
  485. search_query = request.POST["search"]
  486. all_playlists = request.user.playlists.filter(is_in_db=True)
  487. if 'playlist-tags' in request.POST:
  488. tags = request.POST.getlist('playlist-tags')
  489. for tag in tags:
  490. all_playlists = all_playlists.filter(tags__name=tag)
  491. #all_playlists = all_playlists.filter(tags__name__in=tags)
  492. playlist_items = []
  493. if request.POST['search-settings'] == 'starts-with':
  494. playlists = all_playlists.filter(Q(name__istartswith=search_query) | Q(user_label__istartswith=search_query)) if search_query != "" else all_playlists.none()
  495. if search_query != "":
  496. for playlist in all_playlists:
  497. pl_items = playlist.playlist_items.select_related('video').filter(Q(video__name__istartswith=search_query) | Q(video__user_label__istartswith=search_query) & Q(is_duplicate=False))
  498. if pl_items.exists():
  499. for v in pl_items.all():
  500. playlist_items.append(v)
  501. else:
  502. playlists = all_playlists.filter(Q(name__icontains=search_query) | Q(user_label__istartswith=search_query)) if search_query != "" else all_playlists.none()
  503. if search_query != "":
  504. for playlist in all_playlists:
  505. pl_items = playlist.playlist_items.select_related('video').filter(Q(video__name__icontains=search_query) | Q(video__user_label__istartswith=search_query) & Q(is_duplicate=False))
  506. if pl_items.exists():
  507. for v in pl_items.all():
  508. playlist_items.append(v)
  509. return HttpResponse(loader.get_template("intercooler/search_untube_results.html")
  510. .render({"playlists": playlists,
  511. "playlist_items": playlist_items,
  512. "videos_count": len(playlist_items),
  513. "search_query": True if search_query != "" else False,
  514. "all_playlists": all_playlists}))
  515. @login_required
  516. def manage_playlists(request):
  517. return render(request, "manage_playlists.html")
  518. @login_required
  519. def manage_view_page(request, page):
  520. if page == "import":
  521. return render(request, "manage_playlists_import.html",
  522. {"manage_playlists_import_textarea": request.user.profile.manage_playlists_import_textarea})
  523. elif page == "create":
  524. return render(request, "manage_playlists_create.html")
  525. else:
  526. return HttpResponse('Working on this!')
  527. @login_required
  528. @require_POST
  529. def manage_save(request, what):
  530. if what == "manage_playlists_import_textarea":
  531. request.user.profile.manage_playlists_import_textarea = request.POST["import-playlist-textarea"]
  532. request.user.save()
  533. return HttpResponse("")
  534. @login_required
  535. @require_POST
  536. def manage_import_playlists(request):
  537. playlist_links = request.POST["import-playlist-textarea"].replace(",", "").split("\n")
  538. num_playlists_already_in_db = 0
  539. num_playlists_initialized_in_db = 0
  540. num_playlists_not_found = 0
  541. new_playlists = []
  542. old_playlists = []
  543. not_found_playlists = []
  544. done = []
  545. for playlist_link in playlist_links:
  546. if playlist_link.strip() != "" and playlist_link.strip() not in done:
  547. pl_id = Playlist.objects.getPlaylistId(playlist_link.strip())
  548. if pl_id is None:
  549. num_playlists_not_found += 1
  550. continue
  551. status = Playlist.objects.initializePlaylist(request.user, pl_id)["status"]
  552. if status == -1 or status == -2:
  553. print("\nNo such playlist found:", pl_id)
  554. num_playlists_not_found += 1
  555. not_found_playlists.append(playlist_link)
  556. elif status == -3: # playlist already in db
  557. num_playlists_already_in_db += 1
  558. playlist = request.user.playlists.get(playlist_id__exact=pl_id)
  559. old_playlists.append(playlist)
  560. else: # only if playlist exists on YT, so import its videos
  561. print(status)
  562. Playlist.objects.getAllVideosForPlaylist(request.user, pl_id)
  563. playlist = request.user.playlists.get(playlist_id__exact=pl_id)
  564. new_playlists.append(playlist)
  565. num_playlists_initialized_in_db += 1
  566. done.append(playlist_link.strip())
  567. request.user.profile.manage_playlists_import_textarea = ""
  568. request.user.save()
  569. return HttpResponse(loader.get_template("intercooler/manage_playlists_import_results.html")
  570. .render(
  571. {"new_playlists": new_playlists,
  572. "old_playlists": old_playlists,
  573. "not_found_playlists": not_found_playlists,
  574. "num_playlists_already_in_db": num_playlists_already_in_db,
  575. "num_playlists_initialized_in_db": num_playlists_initialized_in_db,
  576. "num_playlists_not_found": num_playlists_not_found
  577. }))
  578. @login_required
  579. @require_POST
  580. def manage_create_playlist(request):
  581. print(request.POST)
  582. return HttpResponse("")
  583. @login_required
  584. def load_more_videos(request, playlist_id, order_by, page):
  585. playlist = request.user.playlists.get(playlist_id=playlist_id)
  586. playlist_items = None
  587. if order_by == "all":
  588. playlist_items = playlist.playlist_items.select_related('video').order_by("video_position")
  589. print(f"loading page 1: {playlist_items.count()} videos")
  590. elif order_by == "favorites":
  591. playlist_items = playlist.playlist_items.select_related('video').filter(video__is_favorite=True).order_by("video_position")
  592. elif order_by == "popularity":
  593. videos = playlist.videos.order_by("-like_count")
  594. elif order_by == "date-published":
  595. videos = playlist.videos.order_by("-published_at")
  596. elif order_by == "views":
  597. videos = playlist.videos.order_by("-view_count")
  598. elif order_by == "has-cc":
  599. videos = playlist.videos.filter(has_cc=True).order_by("video_position")
  600. elif order_by == "duration":
  601. videos = playlist.videos.order_by("-duration_in_seconds")
  602. elif order_by == 'new-updates':
  603. playlist_items = []
  604. if playlist.has_new_updates:
  605. recently_updated_videos = playlist.playlist_items.select_related('video').filter(video__video_details_modified=True)
  606. for playlist_item in recently_updated_videos:
  607. if playlist_item.video.video_details_modified_at + datetime.timedelta(hours=12) < datetime.datetime.now(
  608. pytz.utc): # expired
  609. playlist_item.video.video_details_modified = False
  610. playlist_item.video.save()
  611. if not recently_updated_videos.exists():
  612. playlist.has_new_updates = False
  613. playlist.save()
  614. else:
  615. playlist_items = recently_updated_videos.order_by("video_position")
  616. elif order_by == 'unavailable-videos':
  617. videos = playlist.videos.filter(Q(is_unavailable_on_yt=True) & Q(was_deleted_on_yt=True))
  618. return HttpResponse(loader.get_template("intercooler/videos.html")
  619. .render(
  620. {
  621. "playlist": playlist,
  622. "playlist_items": playlist_items[50 * page:], # only send 50 results per page
  623. "page": page + 1,
  624. "order_by": order_by}))
  625. @login_required
  626. @require_POST
  627. def update_playlist_settings(request, playlist_id):
  628. message_type = "success"
  629. message_content = "Saved!"
  630. print(request.POST)
  631. playlist = request.user.playlists.get(playlist_id=playlist_id)
  632. if "user_label" in request.POST:
  633. playlist.user_label = request.POST["user_label"]
  634. playlist.save(update_fields=['user_label'])
  635. return HttpResponse(loader.get_template("intercooler/messages.html")
  636. .render(
  637. {"message_type": message_type,
  638. "message_content": message_content}))
  639. if 'confirm before deleting' in request.POST:
  640. playlist.confirm_before_deleting = True
  641. else:
  642. playlist.confirm_before_deleting = False
  643. if 'hide videos' in request.POST:
  644. playlist.hide_unavailable_videos = True
  645. else:
  646. playlist.hide_unavailable_videos = False
  647. playlist.save(update_fields=['hide_unavailable_videos', 'confirm_before_deleting'])
  648. valid_title = request.POST['playlistTitle'].replace(">", "greater than").replace("<", "less than")
  649. valid_description = request.POST['playlistDesc'].replace(">", "greater than").replace("<", "less than")
  650. details = {
  651. "title": valid_title,
  652. "description": valid_description,
  653. "privacyStatus": True if request.POST['playlistPrivacy'] == "Private" else False
  654. }
  655. status = Playlist.objects.updatePlaylistDetails(request.user, playlist_id, details)
  656. if status == -1:
  657. message_type = "danger"
  658. message_content = "Could not save :("
  659. return HttpResponse(loader.get_template("intercooler/messages.html")
  660. .render(
  661. {"message_type": message_type,
  662. "message_content": message_content}))
  663. @login_required
  664. def update_playlist(request, playlist_id, type):
  665. playlist = request.user.playlists.get(playlist_id=playlist_id)
  666. if type == "checkforupdates":
  667. print("Checking if playlist changed...")
  668. result = Playlist.objects.checkIfPlaylistChangedOnYT(request.user, playlist_id)
  669. if result[0] == 1: # full scan was done (full scan is done for a playlist if a week has passed)
  670. deleted_videos, unavailable_videos, added_videos = result[1:]
  671. print("CHANGES", deleted_videos, unavailable_videos, added_videos)
  672. # playlist_changed_text = ["The following modifications happened to this playlist on YouTube:"]
  673. if deleted_videos != 0 or unavailable_videos != 0 or added_videos != 0:
  674. pass
  675. # if added_videos > 0:
  676. # playlist_changed_text.append(f"{added_videos} new video(s) were added")
  677. # if deleted_videos > 0:
  678. # playlist_changed_text.append(f"{deleted_videos} video(s) were deleted")
  679. # if unavailable_videos > 0:
  680. # playlist_changed_text.append(f"{unavailable_videos} video(s) went private/unavailable")
  681. # playlist.playlist_changed_text = "\n".join(playlist_changed_text)
  682. # playlist.has_playlist_changed = True
  683. # playlist.save()
  684. else: # no updates found
  685. return HttpResponse("""
  686. <div id="checkforupdates" class="sticky-top" style="top: 0.5em;">
  687. <div class="alert alert-success alert-dismissible fade show visually-hidden" role="alert">
  688. No new updates!
  689. </div>
  690. </div>
  691. """)
  692. elif result[0] == -1: # playlist changed
  693. print("!!!Playlist changed")
  694. # current_playlist_vid_count = playlist.video_count
  695. # new_playlist_vid_count = result[1]
  696. # print(current_playlist_vid_count)
  697. # print(new_playlist_vid_count)
  698. # playlist.has_playlist_changed = True
  699. # playlist.save()
  700. # print(playlist.playlist_changed_text)
  701. else: # no updates found
  702. return HttpResponse("""
  703. <div id="checkforupdates" class="sticky-top" style="top: 0.5em;">
  704. <div class="alert alert-success alert-dismissible fade show visually-hidden sticky-top" role="alert" style="top: 0.5em;">
  705. No new updates!
  706. </div>
  707. </div>
  708. """)
  709. return HttpResponse(f"""
  710. <div hx-get="/playlist/{playlist_id}/update/auto" hx-trigger="load" hx-target="this" class="sticky-top" style="top: 0.5em;">
  711. <div class="alert alert-success alert-dismissible fade show" role="alert">
  712. <div class="d-flex justify-content-center" id="loading-sign">
  713. <img src="/static/svg-loaders/circles.svg" width="40" height="40">
  714. <h5 class="mt-2 ms-2">Changes detected on YouTube, updating playlist '{playlist.name}'...</h5>
  715. </div>
  716. </div>
  717. </div>
  718. """)
  719. if type == "manual":
  720. print("MANUAL")
  721. return HttpResponse(
  722. f"""<div hx-get="/playlist/{playlist_id}/update/auto" hx-trigger="load" hx-swap="outerHTML">
  723. <div class="d-flex justify-content-center mt-4 mb-3" id="loading-sign">
  724. <img src="/static/svg-loaders/circles.svg" width="40" height="40">
  725. <h5 class="mt-2 ms-2">Refreshing playlist '{playlist.name}', please wait!</h5>
  726. </div>
  727. </div>""")
  728. print("Attempting to update playlist")
  729. status, deleted_playlist_item_ids, unavailable_videos, added_videos = Playlist.objects.updatePlaylist(request.user,
  730. playlist_id)
  731. playlist = request.user.playlists.get(playlist_id=playlist_id)
  732. if status == -1:
  733. playlist_name = playlist.name
  734. playlist.delete()
  735. return HttpResponse(
  736. f"""
  737. <div class="d-flex justify-content-center mt-4 mb-3" id="loading-sign">
  738. <h5 class="mt-2 ms-2">Looks like the playlist '{playlist_name}' was deleted on YouTube. It has been removed from UnTube as well.</h5>
  739. </div>
  740. """)
  741. print("Updated playlist")
  742. playlist_changed_text = []
  743. if len(added_videos) != 0:
  744. playlist_changed_text.append(f"{len(added_videos)} added")
  745. for video in added_videos:
  746. playlist_changed_text.append(f"--> {video.name}")
  747. # if len(added_videos) > 3:
  748. # playlist_changed_text.append(f"+ {len(added_videos) - 3} more")
  749. if len(unavailable_videos) != 0:
  750. if len(playlist_changed_text) == 0:
  751. playlist_changed_text.append(f"{len(unavailable_videos)} went unavailable")
  752. else:
  753. playlist_changed_text.append(f"\n{len(unavailable_videos)} went unavailable")
  754. for video in unavailable_videos:
  755. playlist_changed_text.append(f"--> {video.name}")
  756. if len(deleted_playlist_item_ids) != 0:
  757. if len(playlist_changed_text) == 0:
  758. playlist_changed_text.append(f"{len(deleted_playlist_item_ids)} deleted")
  759. else:
  760. playlist_changed_text.append(f"\n{len(deleted_playlist_item_ids)} deleted")
  761. for playlist_item_id in deleted_playlist_item_ids:
  762. playlist_item = playlist.playlist_items.select_related('video').get(playlist_item_id=playlist_item_id)
  763. playlist_changed_text.append(f"--> {playlist_item.video.name}")
  764. playlist_item.delete()
  765. if len(playlist_changed_text) == 0:
  766. playlist_changed_text = ["Successfully refreshed playlist! No new changes found!"]
  767. # return HttpResponse
  768. return HttpResponse(loader.get_template("intercooler/playlist_updates.html")
  769. .render(
  770. {"playlist_changed_text": "\n".join(playlist_changed_text),
  771. "playlist_id": playlist_id}))
  772. @login_required
  773. def view_playlist_settings(request, playlist_id):
  774. try:
  775. playlist = request.user.playlists.get(playlist_id=playlist_id)
  776. except apps.main.models.Playlist.DoesNotExist:
  777. messages.error(request, "No such playlist found!")
  778. return redirect('home')
  779. return render(request, 'view_playlist_settings.html', {"playlist": playlist})
  780. @login_required
  781. def get_playlist_tags(request, playlist_id):
  782. playlist = request.user.playlists.get(playlist_id=playlist_id)
  783. playlist_tags = playlist.tags.all()
  784. return HttpResponse(loader.get_template("intercooler/playlist_tags.html")
  785. .render(
  786. {"playlist_id": playlist_id,
  787. "playlist_tags": playlist_tags}))
  788. @login_required
  789. def get_unused_playlist_tags(request, playlist_id):
  790. playlist = request.user.playlists.get(playlist_id=playlist_id)
  791. user_created_tags = Tag.objects.filter(created_by=request.user)
  792. playlist_tags = playlist.tags.all()
  793. unused_tags = user_created_tags.difference(playlist_tags)
  794. return HttpResponse(loader.get_template("intercooler/playlist_tags_unused.html")
  795. .render(
  796. {"unused_tags": unused_tags}))
  797. @login_required
  798. def get_watch_message(request, playlist_id):
  799. playlist = request.user.playlists.get(playlist_id=playlist_id)
  800. return HttpResponse(loader.get_template("intercooler/playlist_watch_message.html")
  801. .render(
  802. {"playlist": playlist}))
  803. @login_required
  804. @require_POST
  805. def create_playlist_tag(request, playlist_id):
  806. tag_name = request.POST["createTagField"]
  807. if tag_name.lower() == 'Pick from existing unused tags'.lower():
  808. return HttpResponse("Can't use that! Try again >_<")
  809. playlist = request.user.playlists.get(playlist_id=playlist_id)
  810. user_created_tags = Tag.objects.filter(created_by=request.user)
  811. if not user_created_tags.filter(name__iexact=tag_name).exists(): # no tag found, so create it
  812. tag = Tag(name=tag_name, created_by=request.user)
  813. tag.save()
  814. # add it to playlist
  815. playlist.tags.add(tag)
  816. else:
  817. return HttpResponse("""
  818. Already created. Try Again >w<
  819. """)
  820. # playlist_tags = playlist.tags.all()
  821. # unused_tags = user_created_tags.difference(playlist_tags)
  822. return HttpResponse(f"""
  823. Created and Added!
  824. <span class="visually-hidden" hx-get="/playlist/{playlist_id}/get-tags" hx-trigger="load" hx-target="#playlist-tags"></span>
  825. """)
  826. @login_required
  827. @require_POST
  828. def add_playlist_tag(request, playlist_id):
  829. tag_name = request.POST["playlistTag"]
  830. if tag_name == 'Pick from existing unused tags':
  831. return HttpResponse("Pick something! >w<")
  832. playlist = request.user.playlists.get(playlist_id=playlist_id)
  833. playlist_tags = playlist.tags.all()
  834. if not playlist_tags.filter(name__iexact=tag_name).exists(): # tag not on this playlist, so add it
  835. tag = Tag.objects.filter(Q(created_by=request.user) & Q(name__iexact=tag_name)).first()
  836. # add it to playlist
  837. playlist.tags.add(tag)
  838. else:
  839. return HttpResponse("Already Added >w<")
  840. return HttpResponse(f"""
  841. Added!
  842. <span class="visually-hidden" hx-get="/playlist/{playlist_id}/get-tags" hx-trigger="load" hx-target="#playlist-tags"></span>
  843. """)
  844. @login_required
  845. @require_POST
  846. def remove_playlist_tag(request, playlist_id, tag_name):
  847. playlist = request.user.playlists.get(playlist_id=playlist_id)
  848. playlist_tags = playlist.tags.all()
  849. if playlist_tags.filter(name__iexact=tag_name).exists(): # tag on this playlist, remove it it
  850. tag = Tag.objects.filter(Q(created_by=request.user) & Q(name__iexact=tag_name)).first()
  851. print("Removed tag", tag_name)
  852. # remove it from the playlist
  853. playlist.tags.remove(tag)
  854. else:
  855. return HttpResponse("Whoops >w<")
  856. return HttpResponse("")
  857. @login_required
  858. def delete_playlist(request, playlist_id):
  859. playlist = request.user.playlists.get(playlist_id=playlist_id)
  860. if not playlist.is_user_owned: # if playlist trying to delete isn't user owned
  861. playlist.delete() # just delete it from untrue
  862. else:
  863. # delete it from YouTube first then from UnTube
  864. pass
  865. messages.success(request, "Successfully deleted playlist from UnTube.")
  866. return redirect('home')
  867. @login_required
  868. def reset_watched(request, playlist_id):
  869. playlist = request.user.playlists.get(playlist_id=playlist_id)
  870. for video in playlist.videos.filter(Q(is_unavailable_on_yt=False) & Q(was_deleted_on_yt=False)):
  871. video.is_marked_as_watched = False
  872. video.save(update_fields=['is_marked_as_watched'])
  873. # messages.success(request, "Successfully marked all videos unwatched.")
  874. return redirect(f'/playlist/{playlist.playlist_id}')