Python script to get TLE from NASA.

tle_iss_nasa.py

#! /usr/local/bin/python3.6
"""
直近 TLE データ取得 (from NASA)
  : 過去の直近の TLE データ1件を取得する
    （過去データが存在しなければ、未来の直近データ）

  date          name            version
  2018.06.12    mk-mode.com     1.00 新規作成

Copyright(C) 2018 mk-mode.com All Rights Reserved.
---
引数 : [YYYYMMDD[HHMMSS]]
       （JST を指定。無指定なら現在時刻とみなす）
"""
from datetime import datetime
from datetime import timedelta
import re
import requests
import sys
import traceback


class TleIssNasa:
    URL = (
        "https://spaceflight.nasa.gov/realdata/sightings/"
        "SSapplications/Post/JavaSSOP/orbit/ISS/SVPOST.html"
    )
    UA = (
        "mk-mode Bot (by Python/{}.{}.{}, "
        "Administrator: [email protected])"
    ).format(
        sys.version_info.major,
        sys.version_info.minor,
        sys.version_info.micro
    )
    MSG_ERR = (
        "Invalid date!\n"
        "[USAGE] ./tle_iss_nasa.rb [YYYYMMDD[HHMMSS]]"
    )

    def __init__(self):
        if len(sys.argv) < 2:
            self.jst = datetime.now()
        else:
            if re.search(r"^(\d{8}|\d{14})$", sys.argv[1]) is not(None):
                dt = sys.argv[1].ljust(14, "0")
                try:
                    self.jst = datetime.strptime(dt, "%Y%m%d%H%M%S")
                except ValueError as e:
                    print(self.MSG_ERR)
                    sys.exit(1)
            else:
                print(self.MSG_ERR)
                sys.exit(0)
        self.utc = self.jst - timedelta(hours=9)

    def exec(self):
        """ Execution """
        tle = ""
        utc_tle = None
        try:
            print(self.jst.strftime("%Y-%m-%d %H:%M:%S.%f JST"))
            print(self.utc.strftime("%Y-%m-%d %H:%M:%S.%f UTC"))
            print("---")
            tles = self.__get_tle()
            for new in reversed(tles):
                tle = new
                item_utc = re.split(" +", tle[0])[3]
                y = 2000 + int(item_utc[0:2])
                d = float(item_utc[2:])
                utc_tle = datetime(y, 1, 1) + timedelta(days=d)
                if utc_tle <= self.utc:
                    break
            print("\n".join(tle))
            print(utc_tle.strftime("(%Y-%m-%d %H:%M:%S.%f UTC)"))
        except Exception as e:
            raise

    def __get_tle(self):
        """ 最新 TLE 一覧取得 """
        res = []
        try:
            html, status, reason = self.__get_html()
            if status != 200 or reason != "OK":
                print((
                    "STATUS: {} ({})"
                    "[ERROR] Could not retreive html."
                ).format(status, reason))
                sys.exit(1)
            for tle in re.findall(r"ISS\n +(1.+?)\n +(2.+?)\n", html):
                res.append([tle[0], tle[1]])
            return res
        except Exception as e:
            raise

    def __get_html(self):
        """ HTML 取得 """
        try:
            headers = {'User-Agent': self.UA}
            res = requests.get(self.URL, headers)
            return [res.text, res.status_code, res.reason]
        except Exception as e:
            raise


if __name__ == '__main__':
    try:
        obj = TleIssNasa()
        obj.exec()
    except Exception as e:
        traceback.print_exc()
        sys.exit(1)