Setting up more for class etc
This commit is contained in:
Jxck-S
parent
4b3ffeab8e
commit
04379e540a
|
|
@ -0,0 +1,259 @@
|
|||
class Plane:
|
||||
def __init__(self, icao, conf_file):
|
||||
self.icao = icao.upper()
|
||||
self.conf_file = conf_file
|
||||
self.geo_alt_ft = None
|
||||
self.last_geo_alt_ft = None
|
||||
self.last_below_desired_ft = None
|
||||
self.feeding = None
|
||||
self.last_feeding = None
|
||||
self.last_on_ground = None
|
||||
self.on_ground = None
|
||||
self.invalid_Location = None
|
||||
self.longitude = None
|
||||
self.latitude = None
|
||||
self.running_Count = 0
|
||||
self.callsign = None
|
||||
self.takeoff_time = None
|
||||
self.reg = None
|
||||
def getICAO(self):
|
||||
return self.icao
|
||||
def run(self, plane_Dict):
|
||||
#Import Modules
|
||||
#Clear Terminal
|
||||
print("\033[H\033[J")
|
||||
|
||||
#Ability to Remove old Map
|
||||
import os
|
||||
#Setup Geopy
|
||||
from geopy.geocoders import Nominatim
|
||||
geolocator = Nominatim(user_agent="OpenSkyBot", timeout=5)
|
||||
|
||||
import time
|
||||
from colorama import Fore, Back, Style
|
||||
from defOpenSky import pullOpenSky
|
||||
from defADSBX import pullADSBX
|
||||
|
||||
#Setup Config File
|
||||
import configparser
|
||||
config = configparser.ConfigParser()
|
||||
config.read(self.conf_file)
|
||||
|
||||
if config.getboolean('GOOGLE', 'STATICMAP_ENABLE'):
|
||||
from defMap import getMap
|
||||
else:
|
||||
from defSS import getSS
|
||||
|
||||
if config.getboolean('DISCORD', 'ENABLE'):
|
||||
from defDiscord import sendDis
|
||||
#Setup Tweepy
|
||||
if config.getboolean('TWITTER', 'ENABLE'):
|
||||
from defTweet import tweepysetup
|
||||
tweet_api = tweepysetup()
|
||||
#Setup PushBullet
|
||||
if config.getboolean('PUSHBULLET', 'ENABLE'):
|
||||
from pushbullet import Pushbullet
|
||||
pb = Pushbullet(config['PUSHBULLET']['API_KEY'])
|
||||
pb_channel = pb.get_channel(config.get('PUSHBULLET', 'CHANNEL_TAG'))
|
||||
|
||||
#Set Plane ICAO
|
||||
icao = config.get('DATA', 'ICAO').upper()
|
||||
#Pre Set Non Reseting Variables
|
||||
|
||||
running_Count +=1
|
||||
start_time = time.time()
|
||||
print (Back.MAGENTA, "--------", running_Count, "-------------------------------------------------------------", Style.RESET_ALL)
|
||||
#Reset Variables
|
||||
self.below_desired_ft = None
|
||||
self.geo_alt_ft = None
|
||||
self.longitude = None
|
||||
self.latitude = None
|
||||
self.on_ground = None
|
||||
#Get API States for Plane
|
||||
self.plane_Dict = None
|
||||
if config.get('DATA', 'SOURCE') == "OPENS":
|
||||
plane_Dict, failed = pullOpenSky(icao)
|
||||
print (Fore.YELLOW)
|
||||
print ("OpenSky Sourced Data: ", plane_Dict)
|
||||
print(Style.RESET_ALL)
|
||||
elif config.get('DATA', 'SOURCE') == "ADSBX":
|
||||
plane_Dict, failed = pullADSBX(icao)
|
||||
print (Fore.YELLOW)
|
||||
print ("ADSBX Sourced Data: ", plane_Dict)
|
||||
print(Style.RESET_ALL)
|
||||
|
||||
print (Fore.CYAN)
|
||||
print ("Failed:", failed)
|
||||
print ("ICAO:", icao)
|
||||
print(Style.RESET_ALL)
|
||||
|
||||
#Pull Variables from plane_Dict
|
||||
if failed is False:
|
||||
if plane_Dict == None:
|
||||
feeding = False
|
||||
elif plane_Dict != None:
|
||||
locals().update(plane_Dict)
|
||||
print (Fore.CYAN)
|
||||
if config.get('DATA', 'SOURCE') == "ADSBX":
|
||||
print("Registration: ", reg)
|
||||
else:
|
||||
print("Registration: ", "Only shows when using ADSBX!")
|
||||
print ("Callsign: ", callsign)
|
||||
print ("On Ground: ", on_ground)
|
||||
print ("Latitude: ", latitude)
|
||||
print ("Longitude: ", longitude)
|
||||
print ("GEO Alitude Ft: ", geo_alt_ft)
|
||||
print(Style.RESET_ALL)
|
||||
#Lookup Location of coordinates
|
||||
if longitude != None and latitude != None:
|
||||
combined = f"{latitude}, {longitude}"
|
||||
try:
|
||||
location = geolocator.reverse(combined)
|
||||
except:
|
||||
print ("Geopy API Error")
|
||||
print (Fore.YELLOW)
|
||||
# print ("Geopy debug: ", location.raw)
|
||||
print(Style.RESET_ALL)
|
||||
feeding = True
|
||||
else:
|
||||
print (Fore.RED + 'No Location')
|
||||
feeding = False
|
||||
print(Style.RESET_ALL)
|
||||
|
||||
|
||||
|
||||
#Figure if valid location, valid being geopy finds a location
|
||||
if feeding:
|
||||
try:
|
||||
geoError = location.raw['error']
|
||||
except KeyError:
|
||||
invalid_Location = False
|
||||
geoError = None
|
||||
else:
|
||||
invalid_Location = True
|
||||
|
||||
print ("Invalid Location: ", invalid_Location)
|
||||
|
||||
if invalid_Location:
|
||||
print (Fore.RED)
|
||||
print (geoError)
|
||||
print ("Likely Over Water or Invalid Location")
|
||||
print(Style.RESET_ALL)
|
||||
|
||||
|
||||
#Convert Full address to sep variables only if Valid Location
|
||||
elif invalid_Location is False:
|
||||
address = location.raw['address']
|
||||
country = address.get('country', '')
|
||||
country_code = address.get('country_code', '').upper()
|
||||
state = address.get('state', '')
|
||||
county = address.get('county', '')
|
||||
city = address.get('city', '')
|
||||
town = address.get('town', '')
|
||||
hamlet = address.get('hamlet', '')
|
||||
# print (Fore.YELLOW)
|
||||
# print ("Address Fields debug: ", address)
|
||||
# print(Style.RESET_ALL)
|
||||
print (Fore.GREEN)
|
||||
print("Entire Address: ", location.address)
|
||||
print ("Country Code: ", country_code)
|
||||
print ("Country: ", country)
|
||||
print ("State: ", state)
|
||||
print ("City: ", city)
|
||||
print ("Town: ", town)
|
||||
print ("Hamlet: ", hamlet)
|
||||
print ("County: ", county)
|
||||
print(Style.RESET_ALL)
|
||||
|
||||
#Check if below desire ft
|
||||
if geo_alt_ft is None:
|
||||
below_desired_ft = False
|
||||
elif geo_alt_ft < 10000:
|
||||
below_desired_ft = True
|
||||
#Check if tookoff
|
||||
tookoff = bool(invalid_Location is False and below_desired_ft and on_ground is False and ((last_feeding is False and feeding) or (last_on_ground)))
|
||||
print ("Tookoff Just Now:", tookoff)
|
||||
|
||||
|
||||
#Check if Landed
|
||||
landed = bool(last_below_desired_ft and invalid_Location is False and ((last_feeding and feeding is False and last_on_ground is False) or (on_ground and last_on_ground is False)))
|
||||
print ("Landed Just Now:", landed)
|
||||
|
||||
#Chose city town county or hamlet for location as not all are always avalible.
|
||||
if feeding and invalid_Location is False:
|
||||
aera_hierarchy = city or town or county or hamlet
|
||||
#Takeoff Notifcation and Landed
|
||||
if tookoff:
|
||||
tookoff_message = ("Just took off from" + " " + aera_hierarchy + ", " + state + ", " + country_code)
|
||||
print (tookoff_message)
|
||||
#Google Map or tar1090 screenshot
|
||||
if config.getboolean('GOOGLE', 'STATICMAP_ENABLE'):
|
||||
getMap(aera_hierarchy + ", " + state + ", " + country_code)
|
||||
else:
|
||||
getSS(icao)
|
||||
#Discord
|
||||
if config.getboolean('DISCORD', 'ENABLE'):
|
||||
dis_message = config.get('DISCORD', 'TITLE') + " " + tookoff_message
|
||||
sendDis(dis_message)
|
||||
#PushBullet
|
||||
if config.getboolean('PUSHBULLET', 'ENABLE'):
|
||||
with open("map.png", "rb") as pic:
|
||||
map_data = pb.upload_file(pic, "Tookoff IMG")
|
||||
push = pb_channel.push_note(config.get('PUSHBULLET', 'TITLE'), tookoff_message)
|
||||
push = pb_channel.push_file(**map_data)
|
||||
#Twitter
|
||||
if config.getboolean('TWITTER', 'ENABLE'):
|
||||
tweet_api.update_with_media("map.png", status = tookoff_message)
|
||||
takeoff_time = time.time()
|
||||
os.remove("map.png")
|
||||
|
||||
|
||||
if landed:
|
||||
landed_time_msg = ""
|
||||
if takeoff_time != None:
|
||||
landed_time = time.time() - takeoff_time
|
||||
landed_time_msg = time.strftime("Apx. flt. time %H Hours : %M Mins ", time.gmtime(landed_time))
|
||||
landed_message = ("Landed just now in" + " " + aera_hierarchy + ", " + state + ", " + country_code + ". " + landed_time_msg)
|
||||
print (landed_message)
|
||||
#Google Map or tar1090 screenshot
|
||||
if config.getboolean('GOOGLE', 'STATICMAP_ENABLE'):
|
||||
getMap(aera_hierarchy + ", " + state + ", " + country_code)
|
||||
else:
|
||||
getSS(icao)
|
||||
#Discord
|
||||
if config.getboolean('DISCORD', 'ENABLE'):
|
||||
dis_message = config.get('DISCORD', 'TITLE') + " " + landed_message
|
||||
sendDis(dis_message)
|
||||
#PushBullet
|
||||
if config.getboolean('PUSHBULLET', 'ENABLE'):
|
||||
with open("map.png", "rb") as pic:
|
||||
map_data = pb.upload_file(pic, "Landed IMG")
|
||||
push = pb_channel.push_note(config.get('PUSHBULLET', 'TITLE'), landed_message)
|
||||
push = pb_channel.push_file(**map_data)
|
||||
#Twitter
|
||||
if config.getboolean('TWITTER', 'ENABLE'):
|
||||
tweet_api.update_with_media("map.png", status = landed_message)
|
||||
takeoff_time = None
|
||||
landed_time = None
|
||||
time_since_tk = None
|
||||
os.remove("map.png")
|
||||
|
||||
#Set Variables to compare to next check
|
||||
last_feeding = feeding
|
||||
last_geo_alt_ft = geo_alt_ft
|
||||
last_on_ground = on_ground
|
||||
last_below_desired_ft = below_desired_ft
|
||||
|
||||
elif failed:
|
||||
print ("Failed to connect to data source rechecking in 15s")
|
||||
|
||||
if takeoff_time != None:
|
||||
elapsed_time = time.time() - takeoff_time
|
||||
time_since_tk = time.strftime("Time Since Take off %H Hours : %M Mins : %S Secs", time.gmtime(elapsed_time))
|
||||
print(time_since_tk)
|
||||
|
||||
|
||||
|
||||
elapsed_calc_time = time.time() - start_time
|
||||
|
||||
print (Back.MAGENTA, "--------", running_Count, "------------------------Elapsed Time- ", elapsed_calc_time, "-------------------------------------", Style.RESET_ALL)
|
||||
|
|
@ -1,44 +1,41 @@
|
|||
import requests
|
||||
import configparser
|
||||
import json
|
||||
import time
|
||||
from planeClass import Plane
|
||||
main_config = configparser.ConfigParser()
|
||||
main_config.read('mainconf.ini')
|
||||
import os
|
||||
#Set ADSBX URL Based off amount of Conf files
|
||||
if main_config.get('DATA', 'SOURCE') == "ADSBX":
|
||||
planes = {}
|
||||
for filename in os.listdir(os. getcwd()):
|
||||
if filename.endswith(".ini") and filename != "mainconf.ini":
|
||||
plane_config = configparser.ConfigParser()
|
||||
plane_config.read(filename)
|
||||
plane_config.get('DATA', 'ICAO')
|
||||
icao_list = []
|
||||
icao_list.append(plane_config.get('DATA', 'ICAO'))
|
||||
if len(icao_list) > 1:
|
||||
planes[plane_config.get('DATA', 'ICAO').upper()] = Plane(plane_config.get('DATA', 'ICAO'), filename)
|
||||
if len(planes) > 1:
|
||||
url = "https://adsbexchange.com/api/aircraft/json/"
|
||||
elif len(icao_list) == 1:
|
||||
url = "https://adsbexchange.com/api/aircraft/icao/" + icao_list[0].upper() + "/"
|
||||
elif len(planes) == 1:
|
||||
url = "https://adsbexchange.com/api/aircraft/icao/" + next(iter(planes.values())).upper() + "/"
|
||||
|
||||
headers = {
|
||||
'api-auth': main_config.get('ADSBX', 'API_KEY'),
|
||||
'Content-Encoding': 'gzip'
|
||||
}
|
||||
if main_config.get('DATA', 'SOURCE') == "OPENS":
|
||||
|
||||
raise NotImplementedError
|
||||
while True:
|
||||
if main_config.get('DATA', 'SOURCE') == "ADSBX":
|
||||
response = requests.get(url, headers = headers)
|
||||
data = response.text
|
||||
data = json.loads(data)
|
||||
for plane in data['ac']:
|
||||
if plane['icao'] == "4074B8":
|
||||
print(plane)
|
||||
#print(json.dumps(data, indent=2))
|
||||
print("done")
|
||||
#Initate Each Plane Object from their configs
|
||||
for planeData in data:
|
||||
for obj in planes:
|
||||
if planeData['icao'] == obj.getICAO:
|
||||
obj.run(planeData)
|
||||
else:
|
||||
obj.run(None)
|
||||
|
||||
|
||||
|
||||
plane_config = configparser.ConfigParser()
|
||||
plane_config.read(filename)
|
||||
print(filename)
|
||||
print(plane_config.get('ADSBX', 'API_KEY'))
|
||||
time.sleep(15)
|
||||
|
|
|
|||
Loading…
Reference in New Issue