formatting

This commit is contained in:
2023-07-28 18:10:25 +08:00
parent bf82aa62ff
commit 9c26d04d76
2 changed files with 42 additions and 42 deletions

56
firm.py
View File

@@ -78,8 +78,8 @@ class FirmAgent(ap.Agent):
customer.dct_prod_up_prod_stat[
prod]['s_stat'][disrupted_prod][
'set_disrupt_firm'].add(self)
print(f"{self.name} disrupt {customer.name}'s "
f"{prod.code} due to {disrupted_prod.code}")
# print(f"{self.name} disrupt {customer.name}'s "
# f"{prod.code} due to {disrupted_prod.code}")
# remove edge to customer
self.firm_network.graph.remove_edge(n1, n2, key)
@@ -113,12 +113,12 @@ class FirmAgent(ap.Agent):
if status != 'D':
self.dct_prod_up_prod_stat[
prod]['p_stat'].append(('D', self.model.t))
print(f"{self.name}'s {prod.code} turn to D status due to "
f"disrupted supplier of {disrupted_up_prod.code}")
# print(f"{self.name}'s {prod.code} turn to D status due to "
# f"disrupted supplier of {disrupted_up_prod.code}")
def seek_alt_supply(self, product):
# para product is the product that self is seeking
print(f"{self.name} seek alt supply for {product.code}")
# print(f"{self.name} seek alt supply for {product.code}")
if self.dct_n_trial_up_prod_disrupted[
product] <= self.model.int_n_max_trial:
if self.dct_n_trial_up_prod_disrupted[product] == 0:
@@ -171,10 +171,10 @@ class FirmAgent(ap.Agent):
else:
select_alt_supply = \
self.model.nprandom.choice(lst_firm_connect)
print(
f"{self.name} selct alt supply for {product.code} "
f"from {select_alt_supply.name}"
)
# print(
# f"{self.name} selct alt supply for {product.code} "
# f"from {select_alt_supply.name}"
# )
assert select_alt_supply.is_prod_in_current_normal(product), \
f"{select_alt_supply} \
does not produce requested product {product}"
@@ -187,17 +187,17 @@ class FirmAgent(ap.Agent):
select_alt_supply.dct_request_prod_from_firm[product] = [
self
]
print(
select_alt_supply.name, 'dct_request_prod_from_firm', {
key.code: [v.name for v in value]
for key, value in
select_alt_supply.dct_request_prod_from_firm.items()
})
# print(
# select_alt_supply.name, 'dct_request_prod_from_firm', {
# key.code: [v.name for v in value]
# for key, value in
# select_alt_supply.dct_request_prod_from_firm.items()
# })
self.dct_n_trial_up_prod_disrupted[product] += 1
def handle_request(self):
print(self.name, 'handle_request')
# print(self.name, 'handle_request')
for product, lst_firm in self.dct_request_prod_from_firm.items():
if self.dct_prod_capacity[product] > 0:
if len(lst_firm) == 0:
@@ -252,10 +252,10 @@ class FirmAgent(ap.Agent):
down_firm.dct_cand_alt_supp_up_prod_disrupted[
product].remove(self)
print(
f"{self.name} denied {product.code} request "
f"from {down_firm.name} for lack of capacity"
)
# print(
# f"{self.name} denied {product.code} request "
# f"from {down_firm.name} for lack of capacity"
# )
def accept_request(self, down_firm, product):
# para product is the product that self is selling
@@ -288,17 +288,17 @@ class FirmAgent(ap.Agent):
del down_firm.dct_n_trial_up_prod_disrupted[product]
del down_firm.dct_cand_alt_supp_up_prod_disrupted[product]
print(
f"{self.name} accept {product.code} request "
f"from {down_firm.name}"
)
# print(
# f"{self.name} accept {product.code} request "
# f"from {down_firm.name}"
# )
else:
down_firm.dct_cand_alt_supp_up_prod_disrupted[product].remove(self)
print(
f"{self.name} denied {product.code} request "
f"from {down_firm.name}"
)
# print(
# f"{self.name} denied {product.code} request "
# f"from {down_firm.name}"
# )
def clean_before_trial(self):
self.dct_request_prod_from_firm = {}