views.py 45 KB

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