Как сократить код Python

@Gas999 · Регистрация: 12.05.2021

Студворк — интернет-сервис помощи студентам

Добрый день!
Подскажите как можно сократить код, ибо для каждого товара создавать срез - это геморно. Возможно как то через цикл организовать и в случае если наименования товара нет, останавливать цикл.
Если делать по такой схеме, то вы увидите, что переменная "а5" имеет только дату, такого надо избежать.

Помогите пожалуйста я запутался.

Python
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
add = []
test = []
test.append(['32', 'AGA Антифриз Z40', '38', 'Mannol Classic 10W-40', '42', 'Fenom Очиститель карб', '47', 'Gleid Master -30 жид']) 
test.append(['250,00', '380,00', '200,00', '350,00']) 
test.append(['7.00', '22.00', '17.00', '25.00']) 
test.append(['2.00', '500,00', '1.00', '380,00', '1.00', '200,00', '1.00', '350,00']) 
test.append(['06-04-21'])
 
for i in range(1):
    a1 = test[0:5][0][0:2],test[0:5][1][0:1],test[0:5][2][0:1],test[0:5][3][0:2],test[0:5][4][0:1]
    a2 = test[0:5][0][2:4],test[0:5][1][1:2],test[0:5][2][1:2],test[0:5][3][2:4],test[0:5][4][0:1]
    a3 = test[0:5][0][4:6],test[0:5][1][2:3],test[0:5][2][2:3],test[0:5][3][4:6],test[0:5][4][0:1]
    a4 = test[0:5][0][6:8],test[0:5][1][3:4],test[0:5][2][3:4],test[0:5][3][6:8],test[0:5][4][0:1]
    a5 = test[0:5][0][8:10],test[0:5][1][4:5],test[0:5][2][4:5],test[0:5][3][8:10],test[0:5][4][0:1]
 
res1 = []
res2 = []
res3 = []
res4 = []
res5 = []
for x in a1:
    res1.extend(x if isinstance(x, list) else [x])
for x in a2:
    res2.extend(x if isinstance(x, list) else [x])
for x in a3:
    res3.extend(x if isinstance(x, list) else [x])    
for x in a4:
    res4.extend(x if isinstance(x, list) else [x])
for x in a5:
    res5.extend(x if isinstance(x, list) else [x])
 
add.append(res1)
add.append(res2)
add.append(res3)
add.append(res4)
add.append(res5)
print()
print()
print(add)

Вывод:

Code
1
2
3
4
5
[['32', 'AGA Антифриз Z40', '250,00', '7.00', '2.00', '500,00', '06-04-21'],
['38', 'Mannol Classic 10W-40', '380,00', '22.00', '1.00', '380,00', '06-04-21'],
['42', 'Fenom Очиститель карб', '200,00', '17.00', '1.00', '200,00', '06-04-21'],
['47', 'Gleid Master -30 жид', '350,00', '25.00', '1.00', '350,00', '06-04-21'],
['06-04-21']]

@Gas999 · 12.05.2021, 13:50 **[ТС]**

Так же возможно, в решении поможет способ по которому я получаю списки.
Файл чека закрепил.

Код программы:

Python
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
import io
import re
import sqlite3
import sys
import os
from PyQt5 import QtWidgets
from PyQt5.QtWidgets import QApplication, QMainWindow, QPushButton
from PyQt5.uic import loadUi
 
 
class MainWindow(QMainWindow):
    def __init__(self):
        super(MainWindow, self).__init__()
        loadUi("tovar.ui", self)
        self.tableWidget.setColumnWidth(0, 100)
        self.tableWidget.setColumnWidth(1, 100)
        self.tableWidget.setColumnWidth(2, 100)
        self.tableWidget.setColumnWidth(3, 100)
        self.tableWidget.setColumnWidth(4, 100)
        self.tableWidget.setColumnWidth(5, 100)
        self.tableWidget.setColumnWidth(6, 100)
        self.tableWidget.setHorizontalHeaderLabels(["Код", "Наименование товара", "Цена", "Кол-во проданого товара",
                                                    "Сумма продаж товара", "Дата", "Остаток товара"])
        self.loaddata()
        self.btn_target.clicked.connect(work)
        self.btn_folder.clicked.connect(open_folder)
 
    def loaddata(self):
        connection = sqlite3.connect('data1.sqlite')
        cur = connection.cursor()
        sqlstr = 'SELECT * FROM tovar LIMIT 40'
 
        tablerow = 0
        results = cur.execute(sqlstr)
        self.tableWidget.setRowCount(40)
        for row in results:
            self.tableWidget.setItem(tablerow, 0, QtWidgets.QTableWidgetItem(row[0]))
            self.tableWidget.setItem(tablerow, 1, QtWidgets.QTableWidgetItem(row[1]))
            self.tableWidget.setItem(tablerow, 2, QtWidgets.QTableWidgetItem(row[2]))
            self.tableWidget.setItem(tablerow, 3, QtWidgets.QTableWidgetItem(row[3]))
            self.tableWidget.setItem(tablerow, 4, QtWidgets.QTableWidgetItem(row[4]))
            self.tableWidget.setItem(tablerow, 5, QtWidgets.QTableWidgetItem(row[5]))
            self.tableWidget.setItem(tablerow, 6, QtWidgets.QTableWidgetItem(row[6]))
 
            tablerow += 1
 
 
def open_folder():
    path = "C:/"
    path = os.path.realpath(path)
    os.startfile(path)
 
def work():
    # --------------------------------------------------------------------------------------------------
    # ------------------------Тут происходит чтение файла с АЗС и обрезается нужный участок----------------------------
    # --------------------------------------------------------------------------------------------------
    with open('RF105883.004', encoding='cp1251') as f:
        text = f.read()
 
    start, end = -1, -1
    lines = text.splitlines()
    for i, line in enumerate(lines):
        if 'Z - ОТЧЕТ ТОВАРНЫЙ' in line:
            start = i
            continue
 
        if start != -1 and '========================================' in line:
            end = i
            break
 
    new_text = '\n'.join(lines[start: end + 1])
    new_text = new_text.replace('Z - ОТЧЕТ ТОВАРНЫЙ', '')
    new_text = new_text.replace('по кассе # 1', '')
    new_text = new_text.replace('смена #', '')
    new_text = new_text.replace('========================================', '')
    print(new_text)
    new_test = new_text.replace('р.', '')
    with open("isxod.txt", "w") as f:
        f.write(new_test)
 
    str = 'Итого:'
    pattern = re.compile(re.escape(str))
    with open('isxod.txt', 'r+') as f:
        lines = f.readlines()
        f.seek(0)
        for line in lines:
            result = pattern.search(line)
            if result is None:
                f.write(line)
            f.truncate()
 
    # ------------------------[B][COLOR="red"]Срез  кода товара и название товара[/COLOR][/B]
    # ------------------------ВАЖНО: Наименование товара должно быть с точкой в конце, но не в середине!!!
 
    l = []
    a = []
    with open('isxod.txt') as f:
        l = f.read().splitlines()
    del l[0:4]
    s = [x.strip(' ') for x in l]
    str_a = ''.join(s[::4])
    a = str_a.split('.')
    del a[-1]
    f = [[c] for c in a]
    print(f)
 
    # --------------------------[B][COLOR="red"]Срез стоимости товара[/COLOR][/B]
 
    data_price = []
    with io.open('isxod.txt') as myfile:
        for i in myfile.readlines():
            data_price.append(i)
    del data_price[0:5]
    data_j = [x.strip() for x in data_price]
    data_j = list(filter(None, data_j))
    str_j = ''.join(data_j[::4])
    s1 = str_j.replace("р", "")
    lst3 = s1.split()
    b = [[c] for c in lst3]
    print(b)
    #-------------------------- [B][COLOR="red"]Срез Даты из файла[/COLOR][/B]
 
    data_date = []
    with io.open('RF105883.004') as myfile:
        for i in myfile.readlines():
            data_date.append(i)
    del data_date[0:1]
    data_date2 = ''.join(data_date[0])
    lst5 = data_date2.split()
    del lst5[-1]
    print(lst5)
 
    # --------------------------------------------------------------------------------------------------
    # ------------------------Ищет по файлу с АЗС теги и по ним копирует строки в файлы----------
    # --------------------------------------------------------------------------------------------------
 
 
    prod = 'прод:'  # Искомое слово
    ost = 'ост:'
    itog = 'Итого:'
     # ищет и записывает остаток/продажу/итог в файлы
    inp = iter(open('RF105883.004').readlines())
 
    for i in inp:
        if ost in i:
            with open('balance.txt', 'a')as f:
                print(i.replace("ост:", ""), file= f)
 
        if prod in i:
            with open('sell.txt', 'a')as f:
                print(i.replace("прод:", ""), file= f)
 
        if itog in i:
            with open('summ.txt', 'a')as f:
 
                print(i.replace("Итого:", ""), file= f)
 
 
    # --------------------------------------------------------------------------------------------------
    # ------------------------[B][COLOR="Red"]Преобразует строки в файлах и переводит все в массивы[/COLOR][/B]---------------------
    # --------------------------------------------------------------------------------------------------
 
 
    data_ost = []
    with io.open('balance.txt') as myfile:
        for i in myfile.readlines():
            data_ost.append(i)
    data_s = [x.strip() for x in data_ost]
    data_s = list(filter(None, data_s))
    h = [[c] for c in data_s]
    print(h)
 
    data_prod = []
    with io.open('sell.txt') as myfile:
        for i in myfile.readlines():
            data_prod.append(i)
    data_p = [x.strip() for x in data_prod]
    data_p = list(filter(None, data_p))
    str_р = ''.join(data_p)
    s1 = str_р.replace("р.", "")
    lst = s1.split()
    g = [[c] for c in lst]
    print(g)
 
    data_itog = []
    with io.open('summ.txt') as myfile:
        for i in myfile.readlines():
            data_itog.append(i)
    data_i = [x.strip() for x in data_itog]
    data_i = list(filter(None, data_i))
    str_s = ''.join(data_i)
    s2 = str_s.replace("р.", "")
    lst2 = s2.split()
    j = [[c] for c in lst2]
    print(j)
 
    # --------------------------------------------------------------------------------------------------
    # ------------------------Работа с базой данных SQLite-----------------------------
    # --------------------------------------------------------------------------------------------------
 
    conn = sqlite3.connect('data1.sqlite')
    cur = conn.cursor()
    cur.execute("""CREATE TABLE IF NOT EXISTS tovar(
       code TEXT,
       name TEXT,
       price TEXT,
       quantity TEXT,
       summ TEXT,
       data TEXT,
       stock TEXT);
    """)
    conn.commit()
 
    def insert(lineList):
        conn = sqlite3.connect('data1.sqlite')
        c = conn.cursor()
        c.executemany('insert into tovar values (?,?,?,?,?,?,?)', lineList)
        conn.commit()
 
    ar = [['32', 'AGA Антифриз Z40', '250,00', '2.00', '500,00', '06-04-21', '7.00'],
          ['38', 'Mannol Classic 10W-40', '380,00', '1.00', '380,00', '06-04-21', '22.00']]
    insert(ar)
 
 
 
    # --------------------------------------------------------------------------------------------------
    # ------------------------Очищает файлы от содержимого, после обработки-----------------------------
    # --------------------------------------------------------------------------------------------------
 
 
    with open("summ.txt", "w") as f:
        f.write('')
    with open("sell.txt", "w") as f:
        f.write('')
    with open("balance.txt", "w") as f:
        f.write('')
 
 
 
# main
app = QApplication(sys.argv)
mainwindow = MainWindow()
widget = QtWidgets.QStackedWidget()
widget.addWidget(mainwindow)
widget.setFixedHeight(850)
widget.setFixedWidth(1120)
widget.show()
try:
    sys.exit(app.exec_())
except:
    print("Exiting")

@ioprst · 12.05.2021, 14:07

Python
1
2
3
4
5
6
7
8
9
10
11
arr = [[] for _ in range(5)]
n = m = 0
for i in range(5):
    arr[i] = [test[0:5][0][n:n+2], test[0:5][1][m:m+1], test[0:5][2][m:m+1], test[0:5][3][n:n+2], test[0:5][4][0:1]]
    n += 2
    m += 1
 
for i, item in enumerate(arr):
    add.append([])
    for x in item:
        add[i].extend(x if isinstance(x, list) else [x])

@Gas999 · 12.05.2021, 14:16 **[ТС]**

Воу, спасибо большое! Прям работает)

А как дать понять range сколько товара есть при чтении первого списка с переменной 'а1'? Чтоб не создавало лишних строк в двумерный список. Сейчас товара 4, а завтра 12...

@КулХацкеръ · 12.05.2021, 14:40

Python
1
print([list(__import__('itertools').chain(test[0][i*2:i*2+2], test[1][i:i+1], test[2][i:i+1], test[3][i*2:i*2+2], test[4][:])) for i in range(len(test[0]) // 2)])

Количество товара определяется по первой строке массива test.

@ioprst · 12.05.2021, 14:44

Если структура списка не меняется, то можно так

Python
1
2
3
print(len(test[0]) // 2)
# or
print(len(test[1]))

@Gas999 · 13.05.2021, 00:01 **[ТС]**

Спасибо, я нашел ответ.

Товарищ GrAnd вообще все в одну строку вывел, вот пример:

Python
1
2
3
4
5
6
7
8
9
10
test = [['32', 'AGA Антифриз Z40', '38', 'Mannol Classic 10W-40', '42', 'Fenom Очиститель карб', '47', 'Gleid Master -30 жид'],
        ['250,00', '380,00', '200,00', '350,00'],
        ['7.00', '22.00', '17.00', '25.00'],
        ['2.00', '500,00', '1.00', '380,00', '1.00', '200,00', '1.00', '350,00'],
        ['06-04-21']]
 
result = [[el for grp in it for el in grp] + test[-1] for it in zip(*[zip(*[iter(test[i])]*n) for i, n in enumerate([2,1,1,2])])]
 
from pprint import pprint 
pprint(result, width=len(str(result))-1)

ioprst Ваш пример хорош, но вот кол - во товара я могу получить только после цикла, а в range мне изначально нужно указывать кол-во товара.
Вообще способ хорош, возможно для немного других целей. Оставлю себе на заметку!))

КулХацкеръ Ваш пример еще короче, благодарю!))

@Vigi · 13.05.2021, 04:29

Сообщение от Gas999

КулХацкеръ Ваш пример еще короче, благодарю!))

Его пример плох (не соответствует принципам Дзен)!

Новые блоги и статьи Все статьи Все блоги /
Как я обхитрил таблицу Word Alexander-7 21.03.2026 Когда мигает курсор у внешнего края таблицы, и нам надо перейти на новую строку, а при нажатии Enter создается новый ряд таблицы с ячейками, то мы вместо нервных нажатий Энтеров мы пишем любые буквы. . .	Krabik - рыболовный бот для WoW 3.3.5a AmbA 21.03.2026 без регистрации и смс. Это не торговля, приложение не содержит рекламы. Выполняет свою непосредственную задачу - автоматизацию рыбалки в WoW - и ничего более. Однако если админы будут против -. . .	Программный отбор значений справочника Maks 21.03.2026 Установка программного отбора значений справочника "Сотрудники" из модуля формы документа. В качестве фильтра для отбора служит предопределенное значение перечислений. Процедура. . .	Переходник USB-CAN-GPIO Eddy_Em 20.03.2026 Достаточно давно на работе возникла необходимость в переходнике CAN-USB с гальваноразвязкой, оный и был разработан. Однако, все меня терзала совесть, что аж 48-ногий МК используется так тупо: просто. . .
Оттенки серого Argus19 18.03.2026 Оттенки серого Нашёл в интернете 3 прекрасных модуля: Модуль класса открытия диалога открытия/ сохранения файла на Win32 API; Модуль класса быстрого перекодирования цветного изображения в оттенки. . .	SDL3 для Desktop (MinGW): Рисуем цветные прямоугольники с помощью рисовальщика SDL3 на Си и C++ 8Observer8 17.03.2026 Содержание блога Финальные проекты на Си и на C++: finish-rectangles-sdl3-c. zip finish-rectangles-sdl3-cpp. zip	Символические и жёсткие ссылки в Linux. algri14 15.03.2026 Существует два типа ссылок — символические и жёсткие. Ссылка в Linux — это запись в каталоге, которая может указывать либо на inode «файла-ИСТОЧНИКА», тогда это будет «жёсткая ссылка» (hard link),. . .	[Owen Logic] Поддержание уровня воды в резервуаре количеством включённых насосов: моделирование и выбор регулятора ФедосеевПавел 14.03.2026 Поддержание уровня воды в резервуаре количеством включённых насосов: моделирование и выбор регулятора ВВЕДЕНИЕ Выполняя задание на управление насосной группой заполнения резервуара,. . .

@Gas999 0 / 0 / 0 Регистрация: 12.05.2021 Сообщений: 4
	12.05.2021, 14:16 [ТС]
	Воу, спасибо большое! Прям работает) А как дать понять range сколько товара есть при чтении первого списка с переменной 'а1'? Чтоб не создавало лишних строк в двумерный список. Сейчас товара 4, а завтра 12... 0

Как сократить код Python

Решение