From c8f41b97af9e63b14396fd9b323ff16287ea1d63 Mon Sep 17 00:00:00 2001 From: Eriks K Date: Tue, 28 Jul 2020 19:25:22 +0300 Subject: [PATCH] Company cleanup optimisation --- erepublik/classes.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/erepublik/classes.py b/erepublik/classes.py index 1a3e743..dbc1c1a 100644 --- a/erepublik/classes.py +++ b/erepublik/classes.py @@ -244,8 +244,9 @@ class MyCompanies: """ for holding in holdings.values(): if holding.get('id') not in self.holdings: - self.holdings.update( - {int(holding.get('id')): Holding(holding['id'], holding['region_id'], self.citizen)}) + self.holdings.update({ + int(holding.get('id')): Holding(holding['id'], holding['region_id'], self.citizen) + }) if not self.holdings.get(0): self.holdings.update({0: Holding(0, 0, self.citizen)}) # unassigned @@ -293,6 +294,8 @@ class MyCompanies: def __clear_data(self): for holding in self.holdings.values(): + for company in holding.companies: + del company holding.companies.clear() self.companies.clear()