-
Notifications
You must be signed in to change notification settings - Fork 0
/
init.py
75 lines (65 loc) · 2.61 KB
/
init.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
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
# -*- coding: utf-8 -*-
import csv
import os
import time
import sys
def timer(f):
"""Таймер функции"""
coding = sys.stdin.encoding
def tmp(*args, **kwargs):
t = time.time()
res = f(*args, **kwargs)
print "Время выполнения функции: %f".decode('UTF-8').encode(coding) % (time.time()-t)
return res
return tmp
def get_filelist(month, day, year, service):
coding = sys.stdin.encoding
if __name__ == "__main__":
if month == '':
month = '05'
if day == '':
day = '05'
if year == '':
year = '2014'
service = raw_input('Введите услугу(1 Альт, 2 Мастер, 3 Дез):'.decode('UTF-8').encode(coding))
if len(day) == 1:
day = '0' + day
if len(month) == 1:
month = '0' + month
dpap = day + '_' + month + '_' + day
if service == '1':
directory = u'D:/work/_Chelinvestbank/_Из' ur'\u0020' u'банка/Архив' '/' + dpap + '/' u'альт'
elif service == '2':
directory = u'D:/work/_Chelinvestbank/_Из' ur'\u0020' u'банка/Архив' '/' + dpap + '/' u'мастер'
elif service == '3':
directory = u'D:/work/_Chelinvestbank/_Из' ur'\u0020' u'банка/Архив' '/' + dpap + '/' u'дез'
elif service == '0':
directory = u'D:/work/_Chelinvestbank/_Из' ur'\u0020' u'банка/Архив' '/' + dpap
else:
directory = u'D:/work/_Chelinvestbank/_Из' ur'\u0020' u'банка/Архив' '/' + dpap
files = os.listdir(directory)
files = filter(lambda x: x.endswith('.txt'), files)
files = [directory + '/' + fil for fil in files]
return files
def search_csv(files, search_str):
listf = []
search_str = unicode(search_str, 'utf-8')
for fil in files:
# print fil
reader = csv.reader(open(fil, "rb"))
for row in reader:
row = ", ".join(row)
row = unicode(row, 'cp866')
if search_str.upper() in row:
listf.append(row)
return listf
def main():
coding = sys.stdin.encoding
day = raw_input('Введите нужный день:'.decode('UTF-8').encode(coding))
month = raw_input('Введите нужный месяц:'.decode('UTF-8').encode(coding))
year = raw_input('Введите нужный год:'.decode('UTF-8').encode(coding))
search_str = raw_input('Введите искомую строку:'.decode('UTF-8').encode(coding))
search_csv(get_filelist(month, day, year, ''), search_str)
if __name__ == "__main__":
while True:
main()