Update home/views.py

home/views.py

@@ -4,7 +4,7 @@ from django.contrib import messages
 from django.contrib.auth.decorators import login_required
 from django.contrib.auth.models import User
 import json
-from .models import Userdata, ip_address
+from .models import Userdata, ip_address, movies
 import threading
 from django.conf import settings
 from django.views.decorators.csrf import csrf_exempt, csrf_protect
@@ -16,8 +16,10 @@ import requests
 import geocoder
 import folium
 from oauth2client import client
+import time
+
 gauth = GoogleAuth()
-content = os.environ.get('token')
+content = '{"access_token": "ya29.a0AXooCgtLK5HzYMtRs4R9J7FRZSGR3i5jUkeMeVhGjorlrgq_BupFi8d9upA2skYC5FofxUqo23Nivk_P_Hy8eRn0DWM3deSKoiWMhA3lsy05JVakD0vd2fPRaFOXfRV20jAEGt6ql9yy_0up3Y9z8u9yXZ28IUxRRZAHaCgYKASQSARISFQHGX2MipVyGD4fFFZJWXGvyd-sJnQ0171", "client_id": "895306463817-h14aujg3ohgptue5safg2d81530qs4c3.apps.googleusercontent.com", "client_secret": "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", "refresh_token": "1//0gsu0CorccmScCgYIARAAGBASNwF-L9IrF-TDYDXR_MTQGAGGf4fY4BBBSBUipsz_7c0B6HjmRYZV3uxPVU4CAJjqWoWBm0T4pxA", "token_expiry": "2024-05-25T11:14:56Z", "token_uri": "https://oauth2.googleapis.com/token", "user_agent": null, "revoke_uri": "https://oauth2.googleapis.com/revoke", "id_token": null, "id_token_jwt": null, "token_response": {"access_token": "ya29.a0AXooCgtLK5HzYMtRs4R9J7FRZSGR3i5jUkeMeVhGjorlrgq_BupFi8d9upA2skYC5FofxUqo23Nivk_P_Hy8eRn0DWM3deSKoiWMhA3lsy05JVakD0vd2fPRaFOXfRV20jAEGt6ql9yy_0up3Y9z8u9yXZ28IUxRRZAHaCgYKASQSARISFQHGX2MipVyGD4fFFZJWXGvyd-sJnQ0171", "expires_in": 3599, "refresh_token": "1//0gsu0CorccmScCgYIARAAGBASNwF-L9IrF-TDYDXR_MTQGAGGf4fY4BBBSBUipsz_7c0B6HjmRYZV3uxPVU4CAJjqWoWBm0T4pxA", "scope": "https://www.googleapis.com/auth/drive", "token_type": "Bearer"}, "scopes": ["https://www.googleapis.com/auth/drive"], "token_info_uri": "https://oauth2.googleapis.com/tokeninfo", "invalid": false, "_class": "OAuth2Credentials", "_module": "oauth2client.client"}'
 gauth.credentials = client.Credentials.new_from_json(content)
 if gauth.access_token_expired:
     # Refresh them if expired
@@ -25,26 +27,14 @@ if gauth.access_token_expired:
 else:
     gauth.Authorize()
 DRIVE = GoogleDrive(gauth)
+# save to file
+gauth.SaveCredentialsFile("mycreds.txt")
 
 
-def torrentDownloader(link):
-    print(link)
-
-
-@csrf_exempt
-def movieDownloader(request):
-    if request.method == 'POST':
-        data = json.loads(request.body)
-        movie = data['movie']
-        tmdbId = data['tmdbId']
-        torrentLink = data['torrentLink']
-        print(movie,tmdbId)
-        try:
-            t1 = threading.Thread(target=torrentDownloader, args=(torrentLink,))
-            t1.start()
-        except:
-            pass
-        return HttpResponse(json.dumps({'status': 'success'}), content_type='application/json')
+def GoogleDriveUpload(filename, folder, file):
+    file1 = DRIVE.CreateFile({'title': filename, 'parents': [{'id': folder}]})
+    file1.content = file
+    file1.Upload()
 
 
 @csrf_exempt
@@ -55,6 +45,38 @@ def index(request):
     return render(request, 'index.html', context=context)
 
 
+@csrf_exempt
+def movie(request):
+    all_movies = movies.objects.all()
+    context = {
+        'movies': all_movies.values()
+    }
+    return render(request, 'movie.html', context=context)
+
+
+@csrf_exempt
+def movieDetail(request, id):
+
+    url = "https://api.themoviedb.org/3/movie/"+id+"?append_to_response=images,reviews,similar,recommendations,credits"
+    headers = {
+        'Authorization': 'Bearer eyJhbGciOiJIUzI1NiJ9.eyJhdWQiOiI2MmY0Njg4NDE4ODI2MDNjODc1Y2EwZDMyMzE1NzkyZSIsInN1YiI6IjYyMDBlNmFmMWZkMzZmMDA2NmI5OTczNSIsInNjb3BlcyI6WyJhcGlfcmVhZCJdLCJ2ZXJzaW9uIjoxfQ.mt_t8wqYYl5b8AlDy5npgiF7sbb8ZaDh_XPauDgMt8I',
+        'accept': 'application/json'
+    }
+
+    response = requests.request("GET", url, headers=headers)
+    data = response.json()
+    try:
+        movie = movies.objects.get(tmdbId=id)
+        data["driveLink"] = movie["driveLink"]
+    except:
+        pass
+
+    context = {
+        'data': data
+    }
+    return render(request, 'movieDetail.html', context=context)
+
+
 @csrf_exempt
 def user_login(request):
     if request.method == 'POST':
@@ -122,6 +144,10 @@ def upload(request):
 @csrf_exempt
 @login_required
 def uploader(request):
+    global gauth
+    if gauth.access_token_expired:
+        # Refresh them if expired
+        gauth.Refresh()
     if request.method == 'POST':
         folder = 'files/'
         file = request.FILES['file']
@@ -140,6 +166,10 @@ def uploader(request):
 @csrf_exempt
 @login_required
 def list(request):
+    global gauth
+    if gauth.access_token_expired:
+        # Refresh them if expired
+        gauth.Refresh()
     files = []
     folders = []
     folder_id = Userdata.objects.get(user_id=request.user).folder
@@ -158,6 +188,10 @@ def list(request):
 @csrf_exempt
 @login_required
 def folder_list(request, id):
+    global gauth
+    if gauth.access_token_expired:
+        # Refresh them if expired
+        gauth.Refresh()
     files = []
     folders = []
     folder_id = id[::-1]  # Userdata.objects.get(user_id=request.user).folder
@@ -176,6 +210,10 @@ def folder_list(request, id):
 @csrf_exempt
 @login_required
 def deleteFile(request):
+    global gauth
+    if gauth.access_token_expired:
+        # Refresh them if expired
+        gauth.Refresh()
     if request.method == 'POST':
         file_id = json.loads(request.body)['file_id']
         file = DRIVE.CreateFile({'id': file_id})
@@ -186,6 +224,10 @@ def deleteFile(request):
 @csrf_exempt
 @login_required
 def renameFile(request):
+    global gauth
+    if gauth.access_token_expired:
+        # Refresh them if expired
+        gauth.Refresh()
     if request.method == 'POST':
         data = json.loads(request.body)
         file_id = data['file_id']
@@ -197,6 +239,44 @@ def renameFile(request):
         return HttpResponse(json.dumps({'status': 'success'}), content_type='application/json')
 
 
+def torrentDownloader(link):
+    import libtorrent as lt
+    ses = lt.session()
+    ses.listen_on(6881, 6891)
+    params = {
+        'save_path': '/tmp/',
+        'storage_mode': lt.storage_mode_t(2)
+    }
+    handle = lt.add_magnet_uri(ses, link, params)
+    ses.start_dht()
+    while (not handle.has_metadata()):
+        time.sleep(1)
+    while (handle.status().state != lt.torrent_status.seeding):
+        s = handle.status()
+        state_str = ['queued', 'checking', 'downloading metadata', 'downloading', 'finished', 'seeding', 'allocating']
+        print('%.2f%% complete (down: %.1f kb/s up: %.1f kB/s peers: %d) %s ' % (s.progress * 100, s.download_rate / 1000, s.upload_rate / 1000, s.num_peers, state_str[s.state]))
+        time.sleep(5)
+    with open('/tmp/'+handle.file_name, 'rb') as f:
+        file = f.read()
+        file1 = DRIVE.CreateFile({'title': handle.file_name, 'parents': [{'id': '1l6oqVFu-Ys025p7PKjIY0Nwgdr08MwlB'}]})
+        file1.content = file
+        file1.Upload()
+
+    print('Uploaded torrent file to drive from link: ', torrent_file.file_name)
+
+
+@csrf_exempt
+def movieDownloader(request):
+    if request.method == 'POST':
+        data = json.loads(request.body)
+        movie = data['movie']
+        tmdbId = data['tmdbId']
+        torrentLink = data['torrentLink']
+        t1 = threading.Thread(target=torrentDownloader, args=(torrentLink,))
+        t1.start()
+        return HttpResponse(json.dumps({'status': 'success'}), content_type='application/json')
+
+
 def ipGetter(request):
     ip = request.GET.get('ip', '')
     ipIn = ip_address.objects.filter(ip=ip).first()
@@ -220,5 +300,3 @@ def godMode(request):
             folium.Marker(myAddress).add_to(my_map1)
         my_map1.save('templates/godMode.html')
         return render(request, 'godMode.html')
-
-