Replace file_server.sites with file_server.getSites() in ChartCollector.py

This commit is contained in:
Vadim Ushakov 2021-10-22 00:27:47 +07:00
parent c36cba7980
commit 77d2d69376

View file

@ -27,7 +27,7 @@ class ChartCollector(object):
collectors = {} collectors = {}
import main import main
file_server = main.file_server file_server = main.file_server
sites = file_server.sites sites = file_server.getSites()
if not sites: if not sites:
return collectors return collectors
content_db = list(sites.values())[0].content_manager.contents.db content_db = list(sites.values())[0].content_manager.contents.db
@ -102,7 +102,7 @@ class ChartCollector(object):
def getUniquePeers(self): def getUniquePeers(self):
import main import main
sites = main.file_server.sites sites = main.file_server.getSites()
return set(itertools.chain.from_iterable( return set(itertools.chain.from_iterable(
[site.peers.keys() for site in sites.values()] [site.peers.keys() for site in sites.values()]
)) ))
@ -171,7 +171,7 @@ class ChartCollector(object):
collectors = self.getCollectors() collectors = self.getCollectors()
site_collectors = self.getSiteCollectors() site_collectors = self.getSiteCollectors()
import main import main
sites = main.file_server.sites sites = main.file_server.getSites()
i = 0 i = 0
while 1: while 1:
self.collectGlobal(collectors, self.last_values) self.collectGlobal(collectors, self.last_values)