restful.py

# -*- coding: utf-8 -*-
# filename          : restful.py
# description       : API to grab movie links
# author            : Ian Ault
# email             : liketoaccess@protonmail.com
# date              : 05-24-2022
# version           : v1.0
# usage             : python restful.py
# notes             :
# license           : MIT
# py version        : 3.10.2 (must run on 3.6 or higher)
#==============================================================================
import base64
import os
import json
# import threading

from flask_restful import Resource, Api, reqparse
from flask_cors import CORS
from waitress import serve
from flask import Flask

import downloadengine

from settings import *
from scraper import Scraper
from format import Format, contains_only_letters
# from users import User


app = Flask(__name__)
api = Api(app)
CORS(app, resources={r"*": {"origins": "*"}})
scraper = Scraper()
threads = []
# download = Download()


class Queue_Downloads:
	def __init__(self, query, data):
		self.query = query
		self.data = data
		self.queue = []
		self.video_url_list = []
		self.page_link = []

	def setup(self):
		if not isinstance(self.data, list):
			self.queue, status, self.video_url_list, self.page_link = scraper.get_video_url_from_page_link(
				self.query
			)
		else:
			self.video_url_list = []
			for episode_url in self.data:
				result, status, *_, self.page_link = scraper.get_video_url_from_page_link(episode_url)
				self.video_url_list.append(result)
			self.queue = self.data

		if status == 225:
			return self.queue, status, self.video_url_list
			#      results,    225,    remaining_episode_urls

		return self.queue, status, None

	def run(self):
		# print(f"DEBUG: {self.queue}")
		return_codes = []
		for url, data in zip(self.queue, self.video_url_list):
			# self.data is for the Season, NOT for the Episode, the new data needs to be gathered per episode
			print("DEBUG: Starting download...")
			print(f"DEBUG: {data} (data)")
			print(f"DEBUG: {self.page_link} (self.page_link)")
			data = scraper.find_data_from_url(self.page_link)[0]
			print(f"DEBUG: {data} (data)")

			# for index, item in enumerate(data):
			# 	data[index] = item.replace("'", '"')
			# print(data)
			# print(json.loads(data)["title"])

			# download = Download(url, Format(data).format_file_name())
			# download.run()

			filename = Format(data).format_file_name()
			downloadengine.queue.append({"url": url, "filename": filename})
			print("Download queued...")
			print(f"DEBUG: {downloadengine.queue}")

			for index in range(len(downloadengine.queue)):
				downloadengine.download_file(index)

			print("DEBUG: Download finished!")
			# Remove the item from the queue after the download is finished
			downloadengine.queue.remove({"url": url, "filename": filename})

			if DEBUG_MODE:
				return_codes.append([
					{"message": "Created", "data": data}, 201
				])
				continue

			# if the media is a TV Show than the given data is useless
			# return_codes.append(
			# 	[
			# 		{"message": "Created", "data": data}, 201
			# 	] if download.verify() else [
			# 		{"message": "Gone (failure to verify)"}, 410
			# 	]
			# )
			return_codes.append([{"message": "Created", "data": data}, 201])

		return return_codes

class Search(Resource):
	def get(self):
		# Gets search results from a search term
		parser = reqparse.RequestParser()
		parser.add_argument("query", required=True, type=str, location="args")
		args = parser.parse_args()
		if not args:
			return {"message": "Bad request"}, 400

		data = scraper.search(
				args["query"]
			)

		if data == 404 or not data:
			return {"message": "Page not found"}, 404
		return {"message": data}, 200

	def post(self):
		# Adds media to the server
		parser = reqparse.RequestParser()
		parser.add_argument("query", required=True,  type=str, location="args")
		parser.add_argument("data", required=False, type=str, location="args")
		parser.add_argument("remaining_episode_urls", required=False, type=str, location="json")
		args = parser.parse_args()
		if not args:
			return {"message": "Bad request"}, 400

		remaining_episode_urls = None
		if "remaining_episode_urls" in args:
			if args["remaining_episode_urls"]:
				remaining_episode_urls = json.loads(args["remaining_episode_urls"].replace("'" ,'"'))
				print(f"DEBUG: {json.dumps(remaining_episode_urls, indent=4)} (remaining_episode_urls)")

		qd = Queue_Downloads(args["query"], remaining_episode_urls if remaining_episode_urls else args["data"])
		_, status, video_url_list = qd.setup()  # "_" is "queue" (completed results)

		if status == 404:
			return {"message": "Page not found"}, 404
		if status == 225:
			image_data = base64.b64encode(open("captcha.png", "rb").read()).decode("utf-8")
			image_data = f"data:image/png;base64,{image_data}"
			return {"message": "CAPTCHA", "data": image_data, "remaining_episode_urls": video_url_list}, 225

		response_codes = qd.run()
		print(f"DEBUG: {response_codes[-1]}")
		return response_codes[-1][0], response_codes[-1][1]

class Test(Resource):
	def get(self):
		return {"message": "Not implemented"}, 501

	def post(self):
		return {"message": "Not implemented"}, 501

class Catagory(Resource):
	def get(self):
		parser = reqparse.RequestParser()
		parser.add_argument("query", required=True, type=str, location="args")
		args = parser.parse_args()
		if not args:
			return {"message": "Bad request"}, 400

		query = args["query"]
		return {"message": query}, 200

	def post(self):
		return {"message": "Not implemented"}, 501

class Captcha(Resource):
	def get(self):
		if not os.path.exists("captcha.png"):
			return {"message": "File not found"}, 404
		image_data = base64.b64encode(open("captcha.png", "rb").read()).decode("utf-8")
		image_data = f"data:image/png;base64,{image_data}"
		return {"message": "CAPTCHA", "data": image_data}, 225

	def post(self):
		parser = reqparse.RequestParser()
		parser.add_argument("key", required=True, type=str, location="args")
		parser.add_argument("result", required=False, type=str, location="json")
		args = parser.parse_args()
		if not args:
			return {"message": "Bad request"}, 400

		key = args["key"]
		if not contains_only_letters(key):
			return {"message": "Bad request (CAPTCHA contains only uppercase letters)"}, 400

		# if args["result"]:
		# 	result = json.loads(args["result"].replace("'" ,'"'))
		# 	print(json.dumps(result, indent=4))

		captcha_is_resolved = scraper.resolve_captcha(key)
		print(f"CAPTCHA is solved? ({captcha_is_resolved})")
		if not captcha_is_resolved:
			return {"message": "Bad request (CAPTCHA is incorrect)"}, 400
		return {"message": "OK (CAPTCHA resolved)"}, 200

# class Login(Resource):
# 	def get(self):
# 		return {"message": "Not implemented"}, 501

# 	def post(self):
# 		parser = reqparse.RequestParser()
# 		parser.add_argument("token", required=True, type=str, location="args")
# 		args = parser.parse_args()
# 		if not args:
# 			return {"message": "Bad request"}, 400

# 		token = json.loads(
# 			base64.b64decode(
# 				args["token"].split(".")[1] + "=="
# 			).decode("utf-8")
# 		)

# 		user = User(token["email"], token["name"])
# 		if user:
# 			user.modify()
# 			return {"message": "OK (login successful)", "key": user.key}, 200
# 		return {"message": "Forbidden (login does not exist)"}, 403


def main():
	api.add_resource(Search, "/search")
	api.add_resource(Test, "/test")
	api.add_resource(Catagory, "/catagory")
	api.add_resource(Captcha, "/captcha")
	# api.add_resource(Login, "/api/v1/users/login")

	print(f"Starting server on port {PORT}...")

	if DEBUG_MODE:
		print("DEBUG: Running in debug mode")
		app.run(host=HOST, port=PORT, debug=True)
	else:
		serve(app, host=HOST, port=PORT)


if __name__ == "__main__":
	main()