From 3b4b459e7245c7d11197ad1c1d3482cba3e0172d Mon Sep 17 00:00:00 2001 From: Aidar Saifoulline Date: Fri, 15 Oct 2021 21:51:26 +0300 Subject: [PATCH 1/2] add get all platforms function and generator getting platform --- .gitignore | 1 + pyorbital/tlefile.py | 37 +++++++++++++++++++++++++++++++++++++ 2 files changed, 38 insertions(+) diff --git a/.gitignore b/.gitignore index 7a2d2e6d..058f8d23 100644 --- a/.gitignore +++ b/.gitignore @@ -39,3 +39,4 @@ nosetests.xml # rope .ropeproject +/.vs diff --git a/pyorbital/tlefile.py b/pyorbital/tlefile.py index 81a72485..a9ecd095 100644 --- a/pyorbital/tlefile.py +++ b/pyorbital/tlefile.py @@ -105,6 +105,42 @@ def read(platform, tle_file=None, line1=None, line2=None): """ return Tle(platform, tle_file=tle_file, line1=line1, line2=line2) +def get_all_platforms(tle_file=None): + """Read all *platform* name from tle file + tle_file - list path files + """ + platforms = list() + fid = io.open(tle_file, 'rb') + for line in fid: + line = line.decode('utf-8') + line = line.strip() + if not line.startswith('1') and not line.startswith('2') and len(line) > 0: + platforms.append(line) + l_1 = next(fid).decode('utf-8') + l_2 = next(fid).decode('utf-8') + fid.close() + + return platforms + +def get_platform(tle_file=None): + """Read *platform* name from tle file + tle_file - path to tle file + """ + try: + fid = io.open(tle_file, 'rb') + for line in fid: + line = line.decode('utf-8') + line = line.strip() + if not line.startswith('1') and not line.startswith('2') and len(line) > 0: + yield line + l_1 = next(fid).decode('utf-8') + l_2 = next(fid).decode('utf-8') + except IOError: + LOGGER.error("TLE file %s not found.", tle_file) + finally: + if fid: + fid.close() + def fetch(destination): """Fetch TLE from internet and save it to `destination`.""" @@ -213,6 +249,7 @@ def _open(filename): for l_0 in fid: l_0 = l_0.decode('utf-8') if l_0.strip() == self._platform: + if l_0.strip().upper() == self._platform: l_1 = next(fid).decode('utf-8') l_2 = next(fid).decode('utf-8') tle = l_1.strip() + "\n" + l_2.strip() From db89796051c98c95a65a575792a0e13b72139235 Mon Sep 17 00:00:00 2001 From: stickler-ci Date: Fri, 15 Oct 2021 18:54:18 +0000 Subject: [PATCH 2/2] Fixing style errors. --- pyorbital/tlefile.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pyorbital/tlefile.py b/pyorbital/tlefile.py index a9ecd095..b76adb05 100644 --- a/pyorbital/tlefile.py +++ b/pyorbital/tlefile.py @@ -105,6 +105,7 @@ def read(platform, tle_file=None, line1=None, line2=None): """ return Tle(platform, tle_file=tle_file, line1=line1, line2=line2) + def get_all_platforms(tle_file=None): """Read all *platform* name from tle file tle_file - list path files @@ -122,6 +123,7 @@ def get_all_platforms(tle_file=None): return platforms + def get_platform(tle_file=None): """Read *platform* name from tle file tle_file - path to tle file