61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
from localconfig import STORESFILE
|
|
import csv
|
|
|
|
class Stores():
|
|
def __init__(self):
|
|
'''
|
|
Download the business unit CSV from the GK reporting server or pick it up from disk
|
|
|
|
:param self: Description
|
|
'''
|
|
# open csv file to read in all the stores to invoice
|
|
self.stores = {}
|
|
with open(STORESFILE, newline="", encoding="utf-8") as f_in:
|
|
reader = csv.DictReader(f_in)
|
|
for row in reader:
|
|
if row["invoice"] == "1":
|
|
if row['country'] not in self.stores:
|
|
self.stores[row["country"]] = []
|
|
self.stores[row["country"]].append([row["store"], row['name']])
|
|
# self.stores = {}
|
|
# filename = 'businessunitdetails.csv'
|
|
# with open(filename, 'r', encoding='utf-8') as f:
|
|
# while line := f.readline():
|
|
# store, name, dummy = line.split(',',2)
|
|
# self.stores[store] = name
|
|
# print(f" STORES {len(self.stores.keys())}")
|
|
|
|
def all_countries(self):
|
|
return self.stores.keys()
|
|
|
|
def stores_for(self, country):
|
|
result = []
|
|
if country in self.stores.keys():
|
|
for stores in self.stores[country]:
|
|
result.append(stores[0])
|
|
return result
|
|
|
|
def get_store_name(self, store):
|
|
result = '[ store name ]'
|
|
for country in self.stores.keys():
|
|
for s in self.stores[country]:
|
|
if store == s[0]:
|
|
result = s[1]
|
|
return result
|
|
|
|
def report_stores(self):
|
|
result = ''
|
|
for country in self.stores.keys():
|
|
ss = ''
|
|
for s in self.stores[country]:
|
|
ss += ' ' + s[0]
|
|
result += f"{country}:{ss}\n"
|
|
return result
|
|
|
|
def country_for(self, store):
|
|
result = 'XX'
|
|
for country in self.stores.keys():
|
|
for s in self.stores[country]:
|
|
if store == s[0]:
|
|
result = country
|
|
return result |