juliasheats/operate_and_write.py at master · ImHereByChance/juliasheats · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
from parser import *


Calculated = namedtuple('fields', 'costs_singleUse costs_multiUse_deposts costs_multiUse_rents')


def make_codesRates_dict(codes_singleUse:list, rates_singleUse:list):
	"""{codes_codes_singleUse: rates_singleUse, ...etc.}"""
	return dict(zip(codes_singleUse, rates_singleUse))


def filter_repeated_codes(codes_multiUse):
	"""[1, 1, 2, 2] -> [1, 2]"""
	filtred_codes = []
	for code in codes_multiUse:
		if not code in filtred_codes:
			filtred_codes.append(code)
	return filtred_codes


def make_codesPayments_dict(codes_multiUse, deposits_multiUse, rents_multiUse):
	"""codes_multiUse; deposits_multiUse; rents_multiUse ->
	-> {codes_multiUse: (deposits_multiUse, rents_multiUse)}"""
	codes_multiUse = filter_repeated_codes(codes_multiUse)
	depRen_tuples = zip(deposits_multiUse, rents_multiUse)
	codes_and_DepRenTuples = zip(codes_multiUse, depRen_tuples)  #(code_multiUse, (deposits_multiUse, rents_multiUse), ...etc.)
	return dict(codes_and_DepRenTuples)


def calc_packings(data):
	codesRates_dict = make_codesRates_dict(data.codes_singleUse, data.rates_singleUse)
	codesPayments_dict = make_codesPayments_dict(data.codes_multiUse, data.deposits_multiUse,
									  data.rents_multiUse)
	costs_singleUse = []
	costs_multiUse_deposts = []
	costs_multiUse_rents =[]

	index = 0
	for code in data.codes:
		if code in codesRates_dict:
			rate = codesRates_dict[code]
			number = data.numbers[index]

			cost = round(rate * number, 2)

			costs_singleUse.append(cost)
			costs_multiUse_deposts.append(None)
			costs_multiUse_rents.append(None)
			index +=1
		elif code in codesPayments_dict:
			payment = codesPayments_dict[code]
			number = data.numbers[index]

			deposit = payment[0]
			dep_cost = round(deposit * number, 2)

			rent = payment[1]
			rent_cost = round(rent * number, 2)

			costs_multiUse_deposts.append(dep_cost)
			costs_multiUse_rents.append(rent_cost)
			costs_singleUse.append(None)
			index +=1
		else:
			costs_singleUse.append(None)
			costs_multiUse_deposts.append(None)
			costs_multiUse_rents.append(None)
			index +=1

	return Calculated(costs_singleUse, costs_multiUse_deposts, costs_multiUse_rents)


def write_column(sheet, data:list, starting_row, column):
		for i in data:
			sheet.cell(starting_row, column).value = i
			starting_row += 1


def write_results(sample_file, outp_filename, parsed, calculated):
	wb = load_workbook(sample_file)
	sheet = wb.active

	write_column(sheet, parsed.custumers, 5, 4)   		          # Поставщик
	write_column(sheet, parsed.varieties, 5, 5)   		          # Сорт
	write_column(sheet, parsed.prices, 5, 6)      		          # Цена за стебель
	write_column(sheet, parsed.pieces, 5, 7)   		          # Кол-во в коробке
	write_column(sheet, parsed.numbers, 5, 8)  		          # Коробок шт.
	write_column(sheet, parsed.totals, 5, 9)  		          # Всего стеблей
	write_column(sheet, parsed.amounts, 5, 10)  		          # Purchases of products (NE)
	write_column(sheet, calculated.costs_singleUse, 5, 14)  	  # Single use packaging Connect (NE)
	write_column(sheet, calculated.costs_multiUse_deposts, 5, 15)     # Packaging deposit by Connect (AG)
	write_column(sheet, calculated.costs_multiUse_rents, 5, 16)       # Packaging rent by Connect (NE)

	wb.save(outp_filename)

	return outp_filename


if __name__ == '__main__':
	pass