Update home/views.py

home/views.py

@@ -27,27 +27,44 @@ else:
 DRIVE = GoogleDrive(gauth)
 
 
-def GoogleDriveUpload(filename, folder, file):
-    file1 = DRIVE.CreateFile({'title': filename, 'parents': [{'id': folder}]})
-    file1.content = file
-    file1.Upload()
-
 def torrentDownloader(link):
-    import asyncio
-    from torrentp import TorrentDownloader
-    print('Downloading torrent file from link: ', link)
-    torrent_file = TorrentDownloader(link, '/tmp/')
-    # Start the download process
-    asyncio.run(torrent_file.start_download())
-    print('Downloaded torrent file from link: ', link)
-    with open('/tmp/'+torrent_file.file_name, 'rb') as f:
+    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': torrent_file.file_name, 'parents': [{'id': '1l6oqVFu-Ys025p7PKjIY0Nwgdr08MwlB'}]})
+        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')
+
+
 @csrf_exempt
 def movieDownloader(request):
     if request.method == 'POST':