Changed list to set

This commit is contained in:
Petr Kalis 2022-04-19 10:15:16 +02:00
parent 33907e879b
commit 2c11147067
2 changed files with 7 additions and 7 deletions

View file

@ -300,13 +300,13 @@ class SyncServerModule(OpenPypeModule, ITrayModule):
Returns:
(dict): {'site': [alternative sites]...}
"""
alt_site_pairs = defaultdict(list)
alt_site_pairs = defaultdict(set)
for site_name, site_info in conf_sites.items():
alt_sites = set(site_info.get("alternative_sites", []))
alt_site_pairs[site_name].extend(alt_sites)
alt_site_pairs[site_name].update(alt_sites)
for alt_site in alt_sites:
alt_site_pairs[alt_site].append(site_name)
alt_site_pairs[alt_site].add(site_name)
for site_name, alt_sites in alt_site_pairs.items():
sites_queue = deque(alt_sites)
@ -323,7 +323,7 @@ class SyncServerModule(OpenPypeModule, ITrayModule):
alt_alt_site != site_name
and alt_alt_site not in alt_sites
):
alt_sites.append(alt_alt_site)
alt_sites.add(alt_alt_site)
sites_queue.append(alt_alt_site)
return alt_site_pairs

View file

@ -37,9 +37,9 @@ class TestModuleApi(ModuleUnitTest):
"studio2": {"alternative_sites": ["studio"]}}
ret = setup_sync_server_module._get_alt_site_pairs(conf_sites)
expected = {"SFTP": ["studio", "studio2"],
"studio": ["SFTP", "studio2"],
"studio2": ["studio", "SFTP"]}
expected = {"SFTP": {"studio", "studio2"},
"studio": {"SFTP", "studio2"},
"studio2": {"studio", "SFTP"}}
assert ret == expected, "Not matching result"