flight_manager.py

import json
import logging
from io import BytesIO
from typing import Dict, Optional

import requests
from PIL import Image
from telepot.namedtuple import KeyboardButton

import pickle
from config import BASE_URL_RADAR, BASE_URL_MAPS, MAPS_KEY
from db_manager import DbManager


def get_flight_info(user_dict: Dict, txt: str) -> Optional[Dict]:
	volo = txt[3:-1]
	file_da_leggere = open(f"pickle/{user_dict['chat_id']}.pickle", "rb")
	dict_from_file = pickle.load(file_da_leggere)
	try:
		flight_dict = {
			'call': dict_from_file[volo]['Call'],
			'op': dict_from_file[volo]['Op'],
			'from': dict_from_file[volo].get('From'),
			'to': dict_from_file[volo]['To'],
			'hight': round((dict_from_file[volo]['GAlt']) * 0.3048),
			'speed': round((dict_from_file[volo]['Spd']) * 1.852),
			'trak': dict_from_file[volo]['Trak'],
			'velocita_cambio': round((dict_from_file[volo]['Vsi']) * 1.852)
		}

		return flight_dict
	except KeyError as e:
		logging.info('I got a KeyError - reason "%s"' % str(e))
		return None


def elenco_aerei(user_dict: Dict, latitudine, longitudine) -> Optional[Dict]:
	dict_from_file = dict()
	file_da_leggere = open(f"pickle/{user_dict['chat_id']}.pickle", "wb")

	db = DbManager()
	if db.has_quota_reached():
		return {
			'success': False,
			'message': "Limite chiamate mensile raggiunto.",
			'keyboard': [],
			'image': None,
		}
	else:
		url = f"{BASE_URL_RADAR}?lat={latitudine}&lng={longitudine}&fDstL=0&fDstU=100"
		with requests.get(url) as url:
			data = json.loads(url.text)
		db.increase_counter()

	elenco_aerei = []
	markers = ""
	a = 0
	for i in data[u'acList']:
		a += 1
		try:
			markers = markers + "&markers=color:red%7Clabel:" + str(a) + "%7C" + str(i["Lat"]) + "," + str(i["Long"])
			dict_from_file[i['Call']] = i

			elenco_aerei.append(i['Call'])
			logging.info('elenco aerei :' + str(elenco_aerei))
		except KeyError as e:
			logging.info('I got a KeyError - reason "%s"' % str(e))

	new_keyboard = []
	b = 0
	for aereo in elenco_aerei:
		b += 1
		new_keyboard.append([KeyboardButton(text=str(b) + ' >' + aereo + '<')])

	map_url = f"{BASE_URL_MAPS}?center={latitudine},{longitudine}&zoom=9&size=800x800&maptype=roadmap&{markers}" \
			  f"&key={MAPS_KEY}"
	r = requests.get(map_url)
	image_name = f"maps/{user_dict['chat_id']}.png"
	mappa = Image.open(BytesIO(r.content))
	mappa.save(image_name)
	pickle.dump(dict_from_file, file_da_leggere)

	return {
		'success': True,
		'keyboard': new_keyboard,
		'image': image_name,
	}