forked from ivanz4411/school2024-test-task1
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.py
48 lines (33 loc) · 2.27 KB
/
main.py
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
import json
from datetime import datetime
def find_max_month(month):
month_list = [] # Иницализация списка, в котором будут хранится месяцы с максимальной суммой затрат
max_value = 0
month = dict(sorted(month.items(),
key=lambda item: item[1], reverse=True)) # Сортировка словаря по значениям в порядке увеличения
for i in month:
if max_value <= month[i]: # Находим все месяцы с максимальной суммой затрат, добавляем их в список
max_value = month[i]
month_list.append(i)
else:
break
month_list = [datetime.strptime(x, "%m").strftime("%B").lower() for x in
sorted(month_list)] # Сортировка месяцев в порядке их следования, перевод месяцев в нужный формат
return month_list
def report_generation():
with open("format.json", "r") as json_file:
format_dict = json.load(json_file)
month = {} # Инициализация словаря, данные в котором будут {месяц: сумма пользовательских затрат}
for i in format_dict:
if i['status'] == 'COMPLETED': # Проверка условия завершенности заказа
# Из длинного формата даты оставляем только месяц в формате: 01, 02. Инициализируем переменную
ordered_at = datetime.strptime(i["ordered_at"], "%Y-%m-%dT%H:%M:%S.%f").strftime("%m")
if ordered_at in month:
month[ordered_at] += float(i['total'])
else:
month[ordered_at] = float(i['total'])
# Создание словаря в который будет записан ответ. Значение получаем из функции, которая возварщает список месяцев
answer = {"months": find_max_month(month)}
json_string = json.dumps(answer) # Преобразование ответа в json строку
return json_string
report_generation()