GlcoinCore/lndmanage
1
0
Fork 0
mirror of https://github.com/bitromortac/lndmanage.git synced 2026-08-13 12:33:37 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #133 from 3nprob/fix-networkx-pickle-deprecation

Browse source 
network: fix networkx deprecated function usage
This commit is contained in:
bitromortac 2022-05-10 02:12:59 -04:00 • committed by GitHub
commit ebd6431efd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
lndmanage/lib/network.py
View file

@ -62,11 +62,13 @@ class Network:
if timestamp_graph < time.time() - settings.CACHING_RETENTION_MINUTES * 60: # old graph in file
logger.info(f"Cached graph is too old. Fetching new one.")
self.set_graph_and_edges()
nx.write_gpickle(self.graph, cache_graph_filename)
with open(cache_graph_filename, 'wb') as file:
pickle.dump(self.graph, file, pickle.HIGHEST_PROTOCOL)
with open(cache_edges_filename, 'wb') as file:
pickle.dump(self.edges, file)
else: # recent graph in file
self.graph = nx.read_gpickle(cache_graph_filename)
with open(cache_graph_filename, 'rb') as file:
self.graph = pickle.load(file)
with open(cache_edges_filename, 'rb') as file:
self.edges = pickle.load(file)
logger.info(f"> Loaded graph from file: {len(self.graph)} nodes, {len(self.edges)} channels.")