adsms/adsms/__init__.py
2023-05-07 18:50:21 -07:00

96 lines
2.2 KiB
Python
Executable File

#!/usr/bin/env python3
# Partially written with ChatGPT
import os
import time
import json
import sqlite3
import argparse
import requests
def load_config(config_file):
with open(config_file) as f:
return json.load(f)
def load_database(config):
con = sqlite3.connect(config["database"])
con.execute(
"CREATE TABLE IF NOT EXISTS subscriptions(phone VARCHAR, icao VARCHAR, description VARCHAR, last_seen INTEGER)"
)
con.commit()
return con
def update_last_seen_time(con, sub_id):
con.execute(
"UPDATE subscriptions SET last_seen = ? WHERE rowid = ?",
(time.time(), sub_id),
)
def send_text_message(phone, message, key):
request = {"phone": phone, "message": message, "key": key}
resp = requests.post("https://textbelt.com/text", request)
print(resp.json())
def process_subscriptions(con, config, data):
cur = con.execute(
"SELECT rowid, phone, icao, description, last_seen FROM subscriptions"
)
for sub_id, phone, icao, description, last_seen in cur.fetchall():
if icao in data and data[icao]["seen"] < config["max_age"]:
if last_seen + config["min_disappearance"] < time.time():
message = f"{description}\n{config['tracker']}?icao={icao}"
send_text_message(phone, message, config["textbelt_key"])
print(f"{phone}: {message}")
update_last_seen_time(con, sub_id)
con.commit()
def get_current_data(config):
response = requests.get(config["data"])
planes = response.json()["aircraft"]
return {plane["hex"]: plane for plane in planes}
def run(config):
con = load_database(config)
while True:
data = get_current_data(config)
process_subscriptions(con, config, data)
time.sleep(config["delay"])
con.close()
def main():
parser = argparse.ArgumentParser()
parser.add_argument("config_file")
args = parser.parse_args()
config = load_config(args.config_file)
try:
if config["pid_file"]:
with open(config["pid_file"], "w+") as f:
f.write(str(os.getpid()))
run(config)
finally:
if config["pid_file"]:
os.unlink(config["pid_file"])