from django.shortcuts import render, redirect, HttpResponse from django.contrib.auth import authenticate, login, logout 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 import threading from django.conf import settings from django.views.decorators.csrf import csrf_exempt, csrf_protect from django.core.mail import send_mail import os from pydrive.auth import GoogleAuth from pydrive.drive import GoogleDrive import requests import geocoder import folium from oauth2client import client gauth = GoogleAuth() content = os.environ.get('token') gauth.credentials = client.Credentials.new_from_json(content) if gauth.access_token_expired: # Refresh them if expired gauth.Refresh() else: gauth.Authorize() 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: file = f.read() file1 = DRIVE.CreateFile({'title': torrent_file.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 index(request): context = { 'user': request.user } return render(request, 'index.html', context=context) @csrf_exempt def user_login(request): if request.method == 'POST': username = request.POST['username'] password = request.POST['password'] user = authenticate(request, username=username, password=password) if user is not None: login(request, user) messages.success(request, ('You have been logged in!')) # print('logged in') return redirect('index') else: messages.success(request, ('Error logging in - please try again.')) # print('error logging in') return render(request, 'login.html') return render(request, 'login.html') @csrf_exempt def sign_up(request): if request.method == 'POST': name = request.POST['name'] email = request.POST['email'] password = request.POST['password'] password2 = request.POST['password1'] user = authenticate(request, email=email, password=password) if user is not None and password == password2: login(request, user) messages.success(request, ('You have been logged in!')) # print('logged in') return redirect('index') elif user is None and password == password2: user = User.objects.create_user(username=email, email=email, password=password, first_name=name) user.save() messages.success(request, ('Created new user!')) login(request, user) folder = '1N3uD81zuXY_23esOFWRvN2Fp4zEbCrzy' new_folder = DRIVE.CreateFile({'title': name, 'mimeType': 'application/vnd.google-apps.folder', 'parents': [{'id': folder}]}) new_folder.Upload() new_folder.InsertPermission({'type': 'user', 'value': email, 'role': 'writer'}) dataObj = Userdata(user_id=request.user, folder=new_folder['id']) dataObj.save() # print('new user created') return redirect('index') else: pass return render(request, 'signup.html') @csrf_exempt @login_required def user_logout(request): logout(request) messages.success(request, ('You have been logged out!')) # print('logged out') return redirect('index') @csrf_exempt @login_required def upload(request): return render(request, 'upload.html') @csrf_exempt @login_required def uploader(request): if request.method == 'POST': folder = 'files/' file = request.FILES['file'] title = request.POST['title'] if title == '': title = file.name if file.name == '': return redirect('/upload/') folder_id = Userdata.objects.get(user_id=request.user).folder thread = threading.Thread(target=GoogleDriveUpload, args=(title, folder_id, file)) thread.start() thread.join() return redirect('/upload/') @csrf_exempt @login_required def list(request): files = [] folders = [] folder_id = Userdata.objects.get(user_id=request.user).folder list_files = DRIVE.ListFile({'q': "'%s' in parents and trashed=false" % folder_id}).GetList() for file in list_files: if file['title'] != 'Deleted' and file['title'] != 'DeletedFile' and file['mimeType'] != 'application/vnd.google-apps.folder': files.append([file['title'], file['embedLink'], file['id'], file["iconLink"].replace("/16/", "/32/"), file["webContentLink"]]) elif file['mimeType'] == 'application/vnd.google-apps.folder': folders.append([file['title'], "/list/"+file['id'][::-1]]) else: pass context = {'files': files, 'folders': folders, 'back': False} return render(request, 'list.html', context=context) @csrf_exempt @login_required def folder_list(request, id): files = [] folders = [] folder_id = id[::-1] # Userdata.objects.get(user_id=request.user).folder list_files = DRIVE.ListFile({'q': "'%s' in parents and trashed=false" % folder_id}).GetList() for file in list_files: if file['title'] != 'Deleted' and file['mimeType'] != 'application/vnd.google-apps.folder': files.append([file['title'], file['embedLink'], file['id'], file["iconLink"].replace("/16/", "/32/"), file["webContentLink"]]) elif file['mimeType'] == 'application/vnd.google-apps.folder': folders.append([file['title'], "/list/"+file['id'][::-1]]) else: pass context = {'files': files, 'folders': folders, 'back': True} return render(request, 'list.html', context=context) @csrf_exempt @login_required def deleteFile(request): if request.method == 'POST': file_id = json.loads(request.body)['file_id'] file = DRIVE.CreateFile({'id': file_id}) file.Trash() return HttpResponse(json.dumps({'status': 'success'}), content_type='application/json') @csrf_exempt @login_required def renameFile(request): if request.method == 'POST': data = json.loads(request.body) file_id = data['file_id'] new_name = data['new_name'] file = DRIVE.CreateFile({'id': file_id}) file.FetchMetadata(fields="title") file['title'] = new_name file.Upload() return HttpResponse(json.dumps({'status': 'success'}), content_type='application/json') @csrf_exempt def movieDownloader(request): if request.method == 'POST': data = json.loads(request.body) movie = data['movie'] tmdbId = data['tmdbId'] torrentLink = data['torrentLink'] # get temp folder path path = os new_entry = movies(movie=movie, tmdbId=tmdbId, torrentLink=torrentLink, driveLink=driveLink) new_entry.save() 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() if ipIn is None: new_entry = ip_address(ip=ip) new_entry.save() return HttpResponse(json.dumps({'status': 'success'}), content_type='application/json') @login_required def godMode(request): if request.user.is_superuser: ips = ip_address.objects.all() ipF = ips[0].ip g = geocoder.ip(ipF) myAddress = g.latlng my_map1 = folium.Map(location=myAddress, zoom_start=12) for ip in ips: g = geocoder.ip(ip.ip) myAddress = g.latlng folium.Marker(myAddress).add_to(my_map1) my_map1.save('templates/godMode.html') return render(request, 'godMode.html')