scraper.py

import requests
import json
import website
import threading
import time

API_KEY = "tFLMWhfaTXK7"
PROJECT_TOKEN = "tWU6nvGOarc2"
RUN_TOKEN = "t33wTrfmLPF1"


def main(text):

	class Data:
		def __init__(self, api_key, project_token):
			self.API_Key = api_key
			self.Project_Token = project_token
			self.params = {
				"api_key": self.API_Key
			}
			self.data = self.get_data()

		def get_data(self):
			response = requests.get(f'https://www.parsehub.com/api/v2/projects/{self.Project_Token}/last_ready_run/data', params=self.params)
			data = json.loads(response.text)
			return data

		def get_country_info(self, country):
			data = self.data["country"]

			for content in data:
				if content['name'].lower() == country.lower():
					return content

			return "0"


		def get_list(self):
			countries = []
			for country in self.data['country']:
				countries.append(country['name'].lower())

			return countries

		def update_data(self):
			response = requests.post(f'https://www.parsehub.com/api/v2/projects/{self.Project_Token}/run', params=self.params)

			def poll():
				old_data = self.data
				while True:
					new_data = self.get_data()
					if new_data != old_data:
						self.data = new_data
						print("Data updated")
						break


				t = threading.Thread(target=poll)
				t.start()


	data = Data(API_KEY, PROJECT_TOKEN)


	def return_data(country):
		try:
			country_stuff = data.get_country_info(country)
			cases = (country_stuff['total_cases'])
			deaths = (country_stuff['total_deaths'])
			recovered = (country_stuff['total_recovered'])
			active = (country_stuff['active_cases'])
			tests_per_mil = (country_stuff['tests_million'])
			return cases, deaths, recovered, active, tests_per_mil
		except KeyError:
			deaths = 0
			return cases, deaths, recovered, active, tests_per_mil




	data = Data(API_KEY, PROJECT_TOKEN)
	data.update_data()

	user_country = text


	all_data = return_data(user_country)
	return all_data



if __name__ == "__main__":
	main()