46dc675b1f
Note: this was causing a weird flake8 error on CI. Presumably due to CI running flake8 via py3.7. ``` flake8 . --count --select=$ELECTRUM_LINTERS --show-source --statistics ./electrum/scripts/update_default_servers.py:1:26: E999 SyntaxError: invalid syntax #!/usr/bin/env python3 ^ 1 E999 SyntaxError: invalid syntax 1 ```
64 lines
2.3 KiB
Python
Executable File
64 lines
2.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# This script prints a new "servers.json" to stdout.
|
|
# It prunes the offline servers from the existing list (note: run with Tor proxy to keep .onions),
|
|
# and adds new servers from provided file(s) of candidate servers.
|
|
# A file of new candidate servers can be created via e.g.:
|
|
# $ ./electrum/scripts/servers.py > reply.txt
|
|
|
|
import asyncio
|
|
import sys
|
|
import json
|
|
|
|
from electrum.network import Network
|
|
from electrum.util import create_and_start_event_loop, log_exceptions
|
|
from electrum.simple_config import SimpleConfig
|
|
from electrum import constants
|
|
|
|
try:
|
|
fname1 = sys.argv[1]
|
|
fname2 = sys.argv[2] if len(sys.argv) > 2 else None
|
|
except Exception:
|
|
print("usage: update_default_servers.py <file1> [<file2>]")
|
|
print(" - the file(s) should contain json hostmaps for new servers to be added")
|
|
print(" - if two files are provided, their intersection is used (peers found in both).\n"
|
|
" file1 should have the newer data.")
|
|
sys.exit(1)
|
|
|
|
|
|
def get_newly_added_servers(fname1, fname2=None):
|
|
with open(fname1) as f:
|
|
res_hostmap = json.loads(f.read())
|
|
if fname2 is not None:
|
|
with open(fname2) as f:
|
|
dict2 = json.loads(f.read())
|
|
common_set = set.intersection(set(res_hostmap), set(dict2))
|
|
res_hostmap = {k: v for k, v in res_hostmap.items() if k in common_set}
|
|
return res_hostmap
|
|
|
|
|
|
# testnet?
|
|
#constants.set_testnet()
|
|
config = SimpleConfig({'testnet': False})
|
|
|
|
loop, stopping_fut, loop_thread = create_and_start_event_loop()
|
|
network = Network(config)
|
|
network.start()
|
|
|
|
@log_exceptions
|
|
async def f():
|
|
try:
|
|
# prune existing servers
|
|
old_servers_all = constants.net.DEFAULT_SERVERS
|
|
old_servers_online = await network.prune_offline_servers(constants.net.DEFAULT_SERVERS)
|
|
# add new servers
|
|
newly_added_servers = get_newly_added_servers(fname1, fname2)
|
|
res_servers = {**old_servers_online, **newly_added_servers}
|
|
|
|
print(json.dumps(res_servers, indent=4, sort_keys=True))
|
|
print(f"got reply from {len(old_servers_online)}/{len(old_servers_all)} old servers", file=sys.stderr)
|
|
print(f"len(newly_added_servers)={len(newly_added_servers)}. total: {len(res_servers)}", file=sys.stderr)
|
|
finally:
|
|
stopping_fut.set_result(1)
|
|
|
|
asyncio.run_coroutine_threadsafe(f(), loop)
|