views.py 46 KB

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