mirror of
https://github.com/DYefremov/DemonEditor.git
synced 2025-12-22 08:29:41 +01:00
85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
|
import os
|
||
|
|
import shutil
|
||
|
|
from html.parser import HTMLParser
|
||
|
|
|
||
|
|
from app.properties import Profile
|
||
|
|
|
||
|
|
|
||
|
|
class PiconsParser(HTMLParser):
|
||
|
|
""" Parser for html page. (https://www.lyngsat.com/) """
|
||
|
|
|
||
|
|
def __init__(self, entities=False, separator=' '):
|
||
|
|
|
||
|
|
HTMLParser.__init__(self)
|
||
|
|
|
||
|
|
self._parse_html_entities = entities
|
||
|
|
self._separator = separator
|
||
|
|
self._is_td = False
|
||
|
|
self._is_th = False
|
||
|
|
self._current_row = []
|
||
|
|
self._current_cell = []
|
||
|
|
self.rows = []
|
||
|
|
|
||
|
|
def handle_starttag(self, tag, attrs):
|
||
|
|
if tag == 'td':
|
||
|
|
self._is_td = True
|
||
|
|
if tag == 'th':
|
||
|
|
self._is_th = True
|
||
|
|
if tag == "img":
|
||
|
|
self._current_row.append(attrs[0][1])
|
||
|
|
|
||
|
|
def handle_data(self, data):
|
||
|
|
""" Save content to a cell """
|
||
|
|
if self._is_td or self._is_th:
|
||
|
|
self._current_cell.append(data.strip())
|
||
|
|
|
||
|
|
def handle_endtag(self, tag):
|
||
|
|
if tag == 'td':
|
||
|
|
self._is_td = False
|
||
|
|
elif tag == 'th':
|
||
|
|
self._is_th = False
|
||
|
|
|
||
|
|
if tag in ('td', 'th'):
|
||
|
|
final_cell = self._separator.join(self._current_cell).strip()
|
||
|
|
self._current_row.append(final_cell)
|
||
|
|
self._current_cell = []
|
||
|
|
elif tag == 'tr':
|
||
|
|
row = self._current_row
|
||
|
|
ln = len(row)
|
||
|
|
if ln == 10 and row[0].startswith("../logo/"):
|
||
|
|
self.rows.append((row[0], row[-4]))
|
||
|
|
elif ln == 11:
|
||
|
|
self.rows.append((row[0] if row[0].startswith("../logo/") else row[1], row[-4]))
|
||
|
|
self._current_row = []
|
||
|
|
|
||
|
|
def error(self, message):
|
||
|
|
pass
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def parse(open_path, picons_path, tmp_path):
|
||
|
|
with open(open_path, encoding="utf-8", errors="replace") as f:
|
||
|
|
parser = PiconsParser()
|
||
|
|
parser.reset()
|
||
|
|
parser.feed(f.read())
|
||
|
|
rows = parser.rows
|
||
|
|
|
||
|
|
if rows:
|
||
|
|
os.makedirs(picons_path, exist_ok=True)
|
||
|
|
for r in rows:
|
||
|
|
shutil.copyfile(tmp_path + "www.lyngsat.com/" + r[0].lstrip("."),
|
||
|
|
picons_path + PiconsParser.format(r[1], Profile.ENIGMA_2))
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def format(ssid, profile: Profile):
|
||
|
|
if profile is Profile.ENIGMA_2:
|
||
|
|
tr_id = int(ssid[:-2] if len(ssid) < 4 else ssid[:2])
|
||
|
|
return "1_0_1_{:X}_{:X}_{}_1680000_0_0_0.png".format(int(ssid), tr_id, "70")
|
||
|
|
elif profile is Profile.NEUTRINO_MP:
|
||
|
|
return "{:x}{}{:x}".format(int(ssid[:-2]), "0070", int(ssid))
|
||
|
|
else:
|
||
|
|
return "{}.png".format(ssid)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
pass
|