from django.db.models import Q from django.shortcuts import render, redirect from django.contrib.auth import logout from django.views.decorators.http import require_POST from django.contrib.auth.decorators import login_required from allauth.socialaccount.models import SocialToken from django.http import HttpResponse from django.contrib.auth.models import User from django.contrib import messages from apps.main.models import Playlist from .models import Untube from django.template import loader # Create your views here. def index(request): if Untube.objects.all().count() == 0: untube = Untube.objects.create() untube.save() if not request.session.exists(request.session.session_key): request.session.create() request.session['liked_untube'] = False if request.user.is_anonymous: return render(request, 'index.html', {"likes": Untube.objects.all().first().page_likes, "users_joined": User.objects.all().count()}) else: return redirect('home') @login_required def profile(request): user_playlists = request.user.playlists.all() watching = user_playlists.filter(marked_as="watching") total_num_playlists = user_playlists.count() statistics = { "public_x": 0, "private_x": 0, "favorites_x": 0, "watching_x": 0, "imported_x": 0 } if total_num_playlists != 0: # x means percentage statistics["public_x"] = round(user_playlists.filter(is_private_on_yt=False).count() / total_num_playlists, 1) * 100 statistics["private_x"] = round(user_playlists.filter(is_private_on_yt=True).count() / total_num_playlists, 1) * 100 statistics["favorites_x"] = round(user_playlists.filter(is_favorite=True).count() / total_num_playlists, 1) * 100 statistics["watching_x"] = round(user_playlists.filter(marked_as="watching").count() / total_num_playlists, 1) * 100 statistics["imported_x"] = round(user_playlists.filter(is_user_owned=False).count() / total_num_playlists, 1) * 100 return render(request, 'profile.html', { "total_num_playlists": total_num_playlists, "statistics": statistics, "watching": watching}) @login_required def settings(request): return render(request, 'settings.html') @require_POST def update_settings(request): print(request.POST) user = request.user username_input = request.POST['username'].strip() message_content = "Saved!" message_type = "success" if username_input != user.username: if User.objects.filter(username__exact=username_input).count() != 0: message_type = "danger" message_content = f"Username {request.POST['username'].strip()} already taken" else: user.username = request.POST['username'].strip() # user.save() message_content = f"Username updated to {username_input}!" if 'open search in new tab' in request.POST: user.profile.open_search_new_tab = True else: user.profile.open_search_new_tab = False if 'enable gradient bg' in request.POST: user.profile.enable_gradient_bg = True else: user.profile.enable_gradient_bg = False user.save() return HttpResponse(loader.get_template("intercooler/messages.html").render( {"message_type": message_type, "message_content": message_content, "refresh_page": True})) @login_required def delete_account(request): request.user.playlists.all().delete() request.user.videos.all().delete() request.user.playlist_tags.all().delete() request.user.profile.delete() request.user.delete() request.session.flush() messages.success(request, "Account data deleted successfully.") return redirect('index') @login_required def log_out(request): request.session.flush() # delete all stored session keys logout(request) # log out authenticated user if "troll" in request.GET: print("TROLLED") messages.success(request, "Hee Hee") return redirect('/') def cancel_import(request): user_profile = request.user.profile if user_profile.access_token.strip() == "" or user_profile.refresh_token.strip() == "": user_social_token = SocialToken.objects.get(account__user=request.user) user_profile.access_token = user_social_token.token user_profile.refresh_token = user_social_token.token_secret user_profile.expires_at = user_social_token.expires_at # request.user.save() user_profile.imported_yt_playlists = False user_profile.show_import_page = False user_profile.save() return redirect('home') def import_user_yt_playlists(request): request.user.profile.show_import_page = True request.user.profile.save(update_fields=['show_import_page']) return render(request, 'import_in_progress.html') @login_required def start_import(request): """ Initializes only the user's playlist data in the database. Returns the progress bar, which will keep calling continue_import :param request: :return: """ user_profile = request.user.profile if user_profile.access_token.strip() == "" or user_profile.refresh_token.strip() == "": user_social_token = SocialToken.objects.get(account__user=request.user) user_profile.access_token = user_social_token.token user_profile.refresh_token = user_social_token.token_secret user_profile.expires_at = user_social_token.expires_at request.user.save() result = Playlist.objects.initializePlaylist(request.user) if result["status"] == -1: print("User has no YT channel") return HttpResponse(loader.get_template('intercooler/progress_bar.html').render( { "channel_found": False, "error_message": result["error_message"] } )) elif result["status"] == -2: user_profile.import_in_progress = False user_profile.imported_yt_playlists = True user_profile.show_import_page = True user_profile.save() print("User has no playlists on YT") if request.user.profile.yt_channel_id == "": Playlist.objects.getUserYTChannelID(request.user) Playlist.objects.initializePlaylist(request.user, "LL") return HttpResponse(loader.get_template('intercooler/progress_bar.html').render( {"total_playlists": 0, "playlists_imported": 0, "done": True, "progress": 100, "channel_found": True})) else: if request.user.profile.yt_channel_id == "": Playlist.objects.getUserYTChannelID(request.user) Playlist.objects.initializePlaylist(request.user, "LL") user_profile.import_in_progress = True user_profile.save() return HttpResponse(loader.get_template('intercooler/progress_bar.html').render( {"total_playlists": result["num_of_playlists"], "playlist_name": result["first_playlist_name"], "playlists_imported": 0, "progress": 0, "channel_found": True} )) @login_required def continue_import(request): if request.user.profile.import_in_progress is False: return redirect('home') num_of_playlists = request.user.playlists.filter(Q(is_user_owned=True)).exclude(playlist_id="LL").count() print("NUM OF PLAYLISTS", num_of_playlists) try: remaining_playlists = request.user.playlists.filter(Q(is_user_owned=True) & Q(is_in_db=False)).exclude( playlist_id="LL") print(remaining_playlists.count(), "REMAINING PLAYLISTS") playlists_imported = num_of_playlists - remaining_playlists.count() + 1 playlist = remaining_playlists.order_by("created_at")[0] playlist_name = playlist.name playlist_id = playlist.playlist_id Playlist.objects.getAllVideosForPlaylist(request.user, playlist_id) except: print("NO REMAINING PLAYLISTS") playlist_id = -1 if playlist_id != -1: return HttpResponse(loader.get_template('intercooler/progress_bar.html').render( {"total_playlists": num_of_playlists, "playlists_imported": playlists_imported, "playlist_name": playlist_name, "progress": round((playlists_imported / num_of_playlists) * 100, 1), "channel_found": True})) else: # request.user.profile.just_joined = False request.user.profile.import_in_progress = False request.user.profile.imported_yt_playlists = True request.user.profile.show_import_page = True # set back to true again so as to show users the welcome screen on 'home' request.user.save() user_pl_count = request.user.playlists.filter(Q(is_user_owned=True) & Q(is_in_db=True)).exclude( playlist_id="LL").count() return HttpResponse(loader.get_template('intercooler/progress_bar.html').render( {"total_playlists": user_pl_count, "playlists_imported": user_pl_count, "done": True, "progress": 100, "channel_found": True})) @login_required def user_playlists_updates(request, action): """ Gets all user created playlist's ids from YouTube and checks them with the user playlists imported on UnTube. If any playlist id is on UnTube but not on YouTube, deletes the playlist from YouTube. If any new playlist id, imports it to UnTube """ if action == 'check-for-updates': user_playlists_on_UnTube = request.user.playlists.filter(Q(is_user_owned=True) & Q(is_in_db=True)).exclude( playlist_id="LL") result = Playlist.objects.initializePlaylist(request.user) print(result) youtube_playlist_ids = result["playlist_ids"] untube_playlist_ids = [] for playlist in user_playlists_on_UnTube: untube_playlist_ids.append(playlist.playlist_id) deleted_playlist_ids = [] deleted_playlist_names = [] for pl_id in untube_playlist_ids: if pl_id not in youtube_playlist_ids: # ie this playlist was deleted on youtube deleted_playlist_ids.append(pl_id) pl = request.user.playlists.get(playlist_id__exact=pl_id) deleted_playlist_names.append(f"{pl.name} (had {pl.video_count} videos)") pl.delete() if result["num_of_playlists"] == user_playlists_on_UnTube.count() and len(deleted_playlist_ids) == 0: print("No new updates") playlists = [] else: playlists = request.user.playlists.filter(Q(is_user_owned=True) & Q(is_in_db=False)).exclude(playlist_id="LL") print( f"New updates found! {playlists.count()} newly added and {len(deleted_playlist_ids)} playlists deleted!") print(deleted_playlist_names) return HttpResponse(loader.get_template('intercooler/user_playlist_updates.html').render( {"playlists": playlists, "deleted_playlist_names": deleted_playlist_names})) elif action == 'init-update': unimported_playlists = request.user.playlists.filter(Q(is_user_owned=True) & Q(is_in_db=False)).exclude(playlist_id="LL").count() return HttpResponse(f"""
""") elif action == 'start-update': unimported_playlists = request.user.playlists.filter(Q(is_user_owned=True) & Q(is_in_db=False)).exclude(playlist_id="LL") for playlist in unimported_playlists: Playlist.objects.getAllVideosForPlaylist(request.user, playlist.playlist_id) return HttpResponse(""" """) @login_required def get_user_liked_videos_playlist(request): if not request.user.playlists.filter(Q(playlist_id="LL") & Q(is_in_db=True)).exists(): Playlist.objects.initializePlaylist(request.user, "LL") Playlist.objects.getAllVideosForPlaylist(request.user, "LL") messages.success(request, "Successfully imported your Liked Videos playlist!") return HttpResponse(""" """) ### FOR INDEX.HTML @require_POST def like_untube(request): untube = Untube.objects.all().first() untube.page_likes += 1 untube.save() request.session['liked_untube'] = True request.session.save() return HttpResponse(f""" {untube.page_likes} likes (p.s glad you liked it!) """) @require_POST def unlike_untube(request): untube = Untube.objects.all().first() untube.page_likes -= 1 untube.save() request.session['liked_untube'] = False request.session.save() return HttpResponse(f""" {untube.page_likes} likes (p.s :/) """)