Как сократить код 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

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

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

Новые блоги и статьи Все статьи Все блоги /
WordPad для Windows 11 Jel 10.01.2026 WordPad для Windows 11 — это приложение, которое восстанавливает классический текстовый редактор WordPad в операционной системе Windows 11. После того как Microsoft исключила WordPad из. . .	Classic Notepad for Windows 11 Jel 10.01.2026 Old Classic Notepad for Windows 11 Приложение для Windows 11, позволяющее пользователям вернуть классическую версию текстового редактора «Блокнот» из Windows 10. Программа предоставляет более. . .	Почему дизайн решает? Neotwalker 09.01.2026 В современном мире, где конкуренция за внимание потребителя достигла пика, дизайн становится мощным инструментом для успеха бренда. Это не просто красивый внешний вид продукта или сайта — это. . .	Модель микоризы: классовый агентный подход 3 anaschu 06.01.2026 aa0a7f55b50dd51c5ec569d2d10c54f6/ O1rJuneU_ls https:/ / vkvideo. ru/ video-115721503_456239114
Owen Logic: О недопустимости использования связки «аналоговый ПИД» + RegKZR ФедосеевПавел 06.01.2026 Owen Logic: О недопустимости использования связки «аналоговый ПИД» + RegKZR ВВЕДЕНИЕ Введу сокращения: аналоговый ПИД — ПИД регулятор с управляющим выходом в виде числа в диапазоне от 0% до. . .	Модель микоризы: классовый агентный подход 2 anaschu 06.01.2026 репозиторий https:/ / github. com/ shumilovas/ fungi ветка по-частям. коммит Create переделка под биомассу. txt вход sc, но sm считается внутри мицелия. кстати, обьем тоже должен там считаться. . . .	Расчёт токов в цепи постоянного тока igorrr37 05.01.2026 / * Дана цепь постоянного тока с сопротивлениями и напряжениями. Надо найти токи в ветвях. Программа составляет систему уравнений по 1 и 2 законам Кирхгофа и решает её. Последовательность действий:. . .	Новый CodeBlocs. Версия 25.03 palva 04.01.2026 Оказывается, недавно вышла новая версия CodeBlocks за номером 25. 03. Когда-то давно я возился с только что вышедшей тогда версией 20. 03. С тех пор я давно снёс всё с компьютера и забыл. Теперь. . .

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

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

Решение