118 lines
3.3 KiB
Python
118 lines
3.3 KiB
Python
#!/bin/env python3
|
|
import os
|
|
import requests
|
|
import pytz
|
|
from datetime import datetime, timedelta
|
|
|
|
from PIL import Image
|
|
from PIL import ImageFont
|
|
from PIL import ImageDraw
|
|
|
|
from io import BytesIO
|
|
|
|
FONT_SIZE = 40
|
|
TEXT_COLOR = (255, 255, 255, 255)
|
|
|
|
|
|
def invert_back_white(img):
|
|
pixels = img.load()
|
|
for y in range(img.size[1]):
|
|
for x in range(img.size[0]):
|
|
r, g, b, _ = pixels[x, y]
|
|
if abs(r - g) < 5 and abs(r - g) < 5 and abs(b - g) < 5:
|
|
pixels[x, y] = (255 - r, 255 - g, 255 - b, 255)
|
|
|
|
|
|
def scale_to_height(img, new_height):
|
|
width, height = img.size
|
|
new_width = int(new_height * width / height)
|
|
return img.resize((new_width, new_height))
|
|
|
|
|
|
def scale_to_width(img, new_width):
|
|
width, height = img.size
|
|
new_height = int(new_width * height / width)
|
|
return img.resize((new_width, new_height))
|
|
|
|
|
|
def main():
|
|
now = datetime.now(pytz.utc)
|
|
earlier = now - timedelta(hours=1)
|
|
|
|
params = {
|
|
'start': earlier.isoformat(),
|
|
'end': now.isoformat(),
|
|
'status': 'good',
|
|
}
|
|
resp = requests.get('https://network.satnogs.org/api/observations',
|
|
params=params)
|
|
|
|
obs = resp.json()
|
|
obs = reversed(sorted(obs, key=lambda x: x['end']))
|
|
|
|
for ob in obs:
|
|
if ob['waterfall'] is not None:
|
|
break
|
|
|
|
if ob['waterfall'] is None:
|
|
print("No observations with waterfall found")
|
|
return
|
|
|
|
resp = requests.get('https://db.satnogs.org/api/satellites/%s/' %
|
|
ob['norad_cat_id'])
|
|
|
|
sat_name = resp.json()['name']
|
|
sat_img_url = "https://db-satnogs.freetls.fastly.net/media/%s" % (resp.json()['image'])
|
|
print(sat_img_url)
|
|
|
|
mode_desc = ob['transmitter_mode']
|
|
|
|
resp = requests.get(ob['waterfall'])
|
|
waterfall = Image.open(BytesIO(resp.content)).convert('RGBA')
|
|
invert_back_white(waterfall)
|
|
|
|
waterfall = scale_to_height(waterfall, 1024)
|
|
water_width, _ = waterfall.size
|
|
|
|
logo = Image.open("satnogs-logo.png").convert('RGBA')
|
|
logo_w, logo_h = logo.size
|
|
invert_back_white(logo)
|
|
|
|
sat_img = None
|
|
if sat_img_url is not None and sat_img_url != '':
|
|
resp = requests.get(sat_img_url)
|
|
sat_img_aplha = Image.open(BytesIO(resp.content)).convert('RGBA')
|
|
|
|
alpha_chan = sat_img_aplha.convert('RGBA').split()[-1]
|
|
sat_img = Image.new("RGBA", sat_img_aplha.size, (255,255, 255, 255))
|
|
sat_img.paste(sat_img_aplha, mask=alpha_chan)
|
|
|
|
sat_img = scale_to_width(sat_img, 500)
|
|
|
|
|
|
|
|
img = Image.new('RGBA', (1280, 1024))
|
|
draw = ImageDraw.Draw(img)
|
|
draw.rectangle((0, 0, 1280, 1024), fill=(0, 0, 0, 255))
|
|
|
|
station_name = ob['station_name']
|
|
if len(station_name) > 20:
|
|
station_name = station_name[:20] + "..."
|
|
infos = "Satellite: %s\nStation: %s\nMode: %s\nStart: %s\nEnd: %s" % (
|
|
sat_name, station_name, mode_desc, ob['start'], ob['end'])
|
|
|
|
font = ImageFont.truetype("Montserrat-Regular.otf", FONT_SIZE)
|
|
draw.text((water_width + 100, logo_h + 50), infos, TEXT_COLOR, font=font)
|
|
|
|
img.paste(waterfall, (0, 0))
|
|
img.paste(logo, (water_width + 100, 20))
|
|
if sat_img != None:
|
|
img.paste(sat_img, (water_width + 100, 400))
|
|
|
|
img.save('waterfall_tmp.png')
|
|
os.rename('waterfall_tmp.png', 'waterfall.png')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|