mirror of https://github.com/PurpleI2P/pyseeder
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
1.8 KiB
61 lines
1.8 KiB
"""Module for managing transport tasks""" |
|
import urllib.request |
|
from urllib.error import URLError |
|
import os |
|
import importlib |
|
import logging |
|
|
|
from pyseeder.utils import PyseederException, TransportException |
|
|
|
log = logging.getLogger(__name__) |
|
|
|
RESEED_URLS = [ |
|
"https://reseed.i2p-projekt.de/", |
|
"https://i2p.mooo.com/netDb/", |
|
"https://reseed.i2p2.no/", |
|
"https://reseed-fr.i2pd.xyz/", |
|
"https://reseed.memcpy.io/", |
|
"https://reseed.onion.im/", |
|
"https://i2pseed.creativecowpat.net:8443/", |
|
"https://reseed.i2pgit.org/", |
|
"https://i2p.novg.net/", |
|
] |
|
|
|
def download(url, filename): |
|
"""Download .su3 file, return True on success""" |
|
USER_AGENT = "Wget/1.11.4" |
|
|
|
url = "{}i2pseeds.su3".format(url) |
|
req = urllib.request.Request(url, headers={"User-Agent": USER_AGENT}) |
|
|
|
try: |
|
with urllib.request.urlopen(req) as resp: |
|
with open(filename, 'wb') as f: |
|
f.write(resp.read()) |
|
|
|
if os.stat(filename).st_size > 0: |
|
return True |
|
else: |
|
return False |
|
except URLError as e: |
|
return False |
|
|
|
|
|
def upload(filename, config): |
|
"""Upload .su3 file with transports""" |
|
if "transports" in config and "enabled" in config["transports"]: |
|
for t in config["transports"]["enabled"].split(): |
|
if t in config: |
|
tconf = config[t] |
|
else: |
|
tconf = None |
|
|
|
try: |
|
importlib.import_module("pyseeder.transports.{}".format(t)) \ |
|
.run(filename, tconf) |
|
except ImportError: |
|
raise PyseederException( |
|
"{} transport can't be loaded".format(t)) |
|
except TransportException as e: |
|
log.error("Transport error: {}".format(e)) |
|
|
|
|