forked from boramalper/himawaripy
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
This is a huge step in fixing many of the mistakes that we've done in the past and making himawaripy awesome again. * No more `config.py` madness. It was the stupidest mistake I've ever done. Configuration file was unnecessary after all and worse still, it was a native Python script that went into the installation directory, making it impossible for different users sharing the same system have different configs. Now there are only command line arguments. Many thanks to @p4block for raising the issue. * Time offset confusion Maybe it is just me but I think we thought what `hour_offset` was. There is nothing useful (literally) in documentation indicating what it is (except its name). Is it offset from GMT, offset from himawari (which orbits GMT+10:00)? It also caused many errors, such as requesting the image of Earth from the future: boramalper#85, boramalper#82... Now you can either let himawaripy automatically detect it for you or you can give _your offset from GMT_ to let the script fetch right images. * Save battery You might not want himawaripy to drain your battery by downloading 64 tiles in every 10 minutes. Now you can stop himawaripy from refreshing when on battery by passing `--save-battery` argument. Thanks to @vician's suggestion. * Retry on failure Internet sometimes does not just work. Retry up to 3 times for each file before giving up so that not the whole process stops because of a small temporary error. Thanks to @cameronleger. * Output directory Now you can set the output directory for himawaripy if you would like to. ... besides general cleanup, improvements bla bla bla. Sincerely, Bora
- Loading branch information
1 parent
e5d6e05
commit a9fd7f5
Showing
7 changed files
with
271 additions
and
197 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,194 @@ | ||
| #!/usr/bin/env python3 | ||
|
|
||
| import argparse | ||
| from datetime import timedelta, datetime | ||
| import io | ||
| import itertools as it | ||
| import json | ||
| import multiprocessing as mp | ||
| import multiprocessing.dummy as mp_dummy | ||
| import os | ||
| import os.path as path | ||
| import sys | ||
| from time import strptime, strftime, mktime | ||
| import urllib.request | ||
| from glob import iglob, glob | ||
| import threading | ||
| import time | ||
|
|
||
| import appdirs | ||
| from PIL import Image | ||
| from dateutil.tz import tzlocal | ||
|
|
||
| from .utils import set_background, get_desktop_environment | ||
|
|
||
|
|
||
| # Semantic Versioning: Major, Minor, Patch | ||
| HIMAWARIPY_VERSION = (2, 0, 0) | ||
| counter = None | ||
| HEIGHT = 550 | ||
| WIDTH = 550 | ||
|
|
||
|
|
||
| def calculate_time_offset(latest_date, auto, preferred_offset): | ||
| if auto: | ||
| preferred_offset = int(datetime.now(tzlocal()).strftime("%z")[0:3]) | ||
| print("Detected offset: UTC{:+03d}:00".format(preferred_offset)) | ||
| if 11 >= preferred_offset > 10: | ||
| preferred_offset = 10 | ||
| print("Offset is greater than +10, +10 will be used...") | ||
| elif 12 >= preferred_offset > 11: | ||
| preferred_offset = -12 | ||
| print("Offset is greater than +10, -12 will be used...") | ||
|
|
||
| himawari_offset = 10 # UTC+10:00 is the time zone that himawari is over | ||
| offset = int(preferred_offset - himawari_offset) | ||
|
|
||
| offset_tmp = datetime.fromtimestamp(mktime(latest_date)) + timedelta(hours=offset) | ||
| offset_time = offset_tmp.timetuple() | ||
|
|
||
| return offset_time | ||
|
|
||
|
|
||
| def download_chunk(args): | ||
| global counter | ||
|
|
||
| x, y, latest, level = args | ||
| url_format = "http://himawari8.nict.go.jp/img/D531106/{}d/{}/{}_{}_{}.png" | ||
| url = url_format.format(level, WIDTH, strftime("%Y/%m/%d/%H%M%S", latest), x, y) | ||
|
|
||
| tiledata = download(url) | ||
|
|
||
| with counter.get_lock(): | ||
| counter.value += 1 | ||
| if counter.value == level * level: | ||
| print("Downloading tiles: completed.") | ||
| else: | ||
| print("Downloading tiles: {}/{} completed...".format(counter.value, level * level)) | ||
| return x, y, tiledata | ||
|
|
||
|
|
||
| def parse_args(): | ||
| parser = argparse.ArgumentParser(description="set (near-realtime) picture of Earth as your desktop background", | ||
| epilog="http://labs.boramalper.org/himawaripy") | ||
|
|
||
| parser.add_argument("--version", action="version", version="%(prog)s {}.{}.{}".format(*HIMAWARIPY_VERSION)) | ||
|
|
||
| group = parser.add_mutually_exclusive_group() | ||
|
|
||
| group.add_argument("--auto-offset", action="store_true", dest="auto_offset", default=False, | ||
| help="determine offset automatically") | ||
| group.add_argument("-o", "--offset", type=int, dest="offset", default=10, | ||
| help="UTC time offset in hours, must be less than or equal to +10") | ||
|
|
||
| parser.add_argument("-l", "--level", type=int, choices=[4, 8, 16, 20], dest="level", default=4, | ||
| help="increases the quality (and the size) of each tile. possible values are 4, 8, 16, 20") | ||
| parser.add_argument("-d", "--deadline", type=int, dest="deadline", default=6, | ||
| help="deadline in minutes to download all the tiles, set 0 to cancel") | ||
| parser.add_argument("--save-battery", action="store_true", dest="save_battery", default=False, | ||
| help="stop refreshing on battery") | ||
| parser.add_argument("--output-dir", type=str, dest="output_dir", | ||
| help="directory to save the temporary background image", | ||
| default=appdirs.user_cache_dir(appname="himawaripy", appauthor=False)) | ||
|
|
||
| args = parser.parse_args() | ||
|
|
||
| if not -12 <= args.offset <= 10: | ||
| sys.exit("OFFSET has to be between -12 and +10!\n") | ||
|
|
||
| if not args.deadline >= 0: | ||
| sys.exit("DEADLINE has to be greater than (or equal to if you want to disable) zero!\n") | ||
|
|
||
| return args | ||
|
|
||
|
|
||
| def is_discharging(): | ||
| if not sys.platform.startswith("linux"): # I hope this will not end up like Windows 95/98 checks one day... | ||
| sys.exit("Battery saving feature works only on linux!\n") | ||
|
|
||
| if len(glob("/sys/class/power_supply/BAT*")) > 1: | ||
| print("Multiple batteries detected, using BAT0.") | ||
|
|
||
| with open("/sys/class/power_supply/BAT0/status") as f: | ||
| status = f.readline().strip() | ||
|
|
||
| return status == "Discharging" | ||
|
|
||
|
|
||
| def download(url): | ||
| exception = None | ||
|
|
||
| for i in range(1, 4): # retry max 3 times | ||
| try: | ||
| with urllib.request.urlopen(url) as response: | ||
| return response.read() | ||
| except Exception as e: | ||
| exception = e | ||
| print("[{}/3] Retrying to download '{}'...".format(i, url)) | ||
| time.sleep(1) | ||
| pass | ||
|
|
||
| if exception: | ||
| raise exception | ||
| else: | ||
| sys.exit("Could not download '{}'!\n".format(url)) | ||
|
|
||
|
|
||
| def thread_main(args): | ||
| global counter | ||
| counter = mp.Value("i", 0) | ||
|
|
||
| level = args.level # since we are going to use it a lot of times | ||
|
|
||
| print("Updating...") | ||
| latest_json = download("http://himawari8-dl.nict.go.jp/himawari8/img/D531106/latest.json") | ||
| latest = strptime(json.loads(latest_json.decode("utf-8"))["date"], "%Y-%m-%d %H:%M:%S") | ||
|
|
||
| print("Latest version: {} GMT.".format(strftime("%Y/%m/%d %H:%M:%S", latest))) | ||
| requested_time = calculate_time_offset(latest, args.auto_offset, args.offset) | ||
| if args.auto_offset or args.offset != 10: | ||
| print("Offset version: {} GMT.".format(strftime("%Y/%m/%d %H:%M:%S", requested_time))) | ||
|
|
||
| png = Image.new("RGB", (WIDTH * level, HEIGHT * level)) | ||
|
|
||
| p = mp_dummy.Pool(level * level) | ||
| print("Downloading tiles...") | ||
| res = p.map(download_chunk, it.product(range(level), range(level), (requested_time,), (args.level,))) | ||
|
|
||
| for (x, y, tiledata) in res: | ||
| tile = Image.open(io.BytesIO(tiledata)) | ||
| png.paste(tile, (WIDTH * x, HEIGHT * y, WIDTH * (x + 1), HEIGHT * (y + 1))) | ||
|
|
||
| for file in iglob(path.join(args.output_dir, "himawari-*.png")): | ||
| os.remove(file) | ||
|
|
||
| output_file = path.join(args.output_dir, strftime("himawari-%Y%m%dT%H%M%S.png", requested_time)) | ||
| print("Saving to '%s'..." % (output_file,)) | ||
| os.makedirs(path.dirname(output_file), exist_ok=True) | ||
| png.save(output_file, "PNG") | ||
|
|
||
| if not set_background(output_file): | ||
| sys.exit("Your desktop environment '{}' is not supported!\n".format(get_desktop_environment())) | ||
|
|
||
|
|
||
| def main(): | ||
| args = parse_args() | ||
|
|
||
| print("himawaripy {}.{}.{}".format(*HIMAWARIPY_VERSION)) | ||
|
|
||
| if args.save_battery and is_discharging(): | ||
| sys.exit("Discharging!\n") | ||
|
|
||
| main_thread = threading.Thread(target=thread_main, args=(args,), name="himawaripy-main-thread", daemon=True) | ||
| main_thread.start() | ||
| main_thread.join(args.deadline * 60 if args.deadline else None) | ||
|
|
||
| if args.deadline and main_thread.is_alive(): | ||
| sys.exit("Timeout!\n") | ||
|
|
||
| print() | ||
| sys.exit(0) | ||
|
|
||
|
|
||
| if __name__ == "__main__": | ||
| main() |
This file was deleted.
Oops, something went wrong.
Oops, something went wrong.