Scripts/Write-offs.py at main · Kirvio/Scripts · 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
import xml.dom.minidom as minidom
import argparse
import logging
import time
import sys
try:
    from xlsxwriter.workbook import Workbook
except ImportError as err:
    print(err)
    time.sleep(20)

document = 'C:\\PythonProgs\\ForReports\\Write-offs\\DeptFor3Years27072021.xml'

def get_all_text(node) -> str:
    if node.nodeType == node.TEXT_NODE:
        value = node.data
    else:
        value = ''.join(get_all_text(child_node) for child_node in node.childNodes)

    return value

def ParseXML(xml):
    try:
        with minidom.parse(xml) as doc:
            xml_tuple = (
                "col_login", "col_account_id", "col_start_time"
            )
            y = []
            for row_el in doc.getElementsByTagName("row"):
                row = []
                for tag_name in xml_tuple:
                    child_el = row_el.getElementsByTagName(tag_name)[0]
                    child_text = get_all_text(child_el) or "No data"
                    row.append(child_text)

                y.append(row)
    except (Exception, TypeError, AttributeError):
        log.error("Exception occurred", exc_info=True)
        time.sleep(20)
    else:
        log.info("Ищем должников за текущий месяц")
        yield tuple(y)

def ParseForMonth(xml, month, year):
    try:
        data = next(ParseXML(xml))
        __month_dict = {
            '01': 'Jan', '02': 'Feb',
            '03': 'Mar', '04': 'Apr',
            '05': 'May', '06': 'Jun',
            '07': 'Jul', '08': 'Aug',
            '09': 'Sep', '10': 'Oct',
            '11': 'Nov', '12': 'Dec'
        }

        x = [item for item in data \
                                   if __month_dict[month] in item[2] \
                                                                     and year in item[2]]

        log.info(f"Количество должников за {'.'.join([month, year])}: {len(x)}")
    except (Exception, TypeError, AttributeError):
        log.error("Exception occurred", exc_info=True)
        time.sleep(20)
    else:
        try:
            with Workbook('C:\\PythonProgs\\ForReports\\Write-offs\\Списания.xlsx') as workbook:
                worksheet = workbook.add_worksheet()
                for row_num, row_data in enumerate(x):
                    for col_num, col_data in enumerate(row_data):
                        worksheet.set_column(col_num, 5, 20)
                        worksheet.write(row_num+1, col_num, col_data)
        except (Exception, TypeError, AttributeError):
            log.error("Exception occurred", exc_info=True)
            time.sleep(20)
        else:
            log.info("Экспортировано в xlsx успешно")
            input("Нажмите Enter для выхода.")

if __name__ == "__main__":
    try:
        log = logging.getLogger(__name__)
        log.setLevel(logging.INFO)
        f_format = logging.Formatter('%(asctime)s %(levelname)s %(message)s')
        handler = logging.StreamHandler(sys.stdout)
        handler.setLevel(logging.INFO)
        handler.setFormatter(f_format)
        log.addHandler(handler)
    except Exception as exc:
        print(exc)
        time.sleep(20)
    else:
        try:
            parser = argparse.ArgumentParser(
                description='Скрипт для поиска должников в xml файлике',
            )
            parser.add_argument("--month", help="Введите дату(месяц)", required=True)
            parser.add_argument("--year", help="Введите дату(год)", required=True)
        except Exception as exc:
            print(exc)
            time.sleep(20)
        else:
            args = parser.parse_args()
            ParseForMonth(xml=document, month=args.month, year=args.year)