-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathhaode_m_m_3.py
More file actions
129 lines (126 loc) · 5.13 KB
/
haode_m_m_3.py
File metadata and controls
129 lines (126 loc) · 5.13 KB
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
#coding=gbk
"""
demo:
python haode_m_m_3.py abs:output_path abs:input_path(haode_m.yyyymmdd) abs:input_dir startDate(yyyymmdd) endDate(yyymmdd)
"""
from xlwt import *
import sys
import logging
if __name__ == '__main__':
if len(sys.argv) != 6:
logging.error("""
Usage: python input_name output_name
demo
python haode_m_m_3.py abs:output_path abs:input_path(haode_m.yyyymmdd) abs:input_dir startDate(yyyymmdd) endDate(yyymmdd)
""")
exit(1)
save_file_path = sys.argv[1]
check_file_path = sys.argv[2]
m_dir = sys.argv[3]
import time
startDate = time.strptime(sys.argv[4], '%Y%m%d')
endDate = time.strptime(sys.argv[5], "%Y%m%d")
prefix = time.strftime("%y.%m.%d", startDate) + "-" + time.strftime("%y.%m.%d", endDate)
headings = [u'店名', u'编号', u'交易金额']
book = Workbook(encoding='cp1251')
book.add_sheet(u'差异调整表')
fontA = Font()
fontA.name = u'宋体'
fontA.height = 220
styleA = XFStyle()
styleA.font = fontA
fontB = Font()
fontB.name = u'宋体'
fontB.bold = True
fontB.height = 220
styleB = XFStyle()
styleB.font = fontB
import codecs
"""
month to days summary
"""
try:
import datetime
sd = datetime.datetime(startDate.tm_year,startDate.tm_mon, startDate.tm_mday)
ed = datetime.datetime(endDate.tm_year, endDate.tm_mon, endDate.tm_mday)
m_total_data = []
import os
while sd <= ed:
d_file = os.path.join(m_dir, 'haode_d.{sdtime}'.format(sdtime = sd.strftime("%Y%m%d")))
if os.path.exists(d_file):
logging.debug("parse ..ing " + d_file)
fhm = codecs.open(filename=d_file, mode='r', encoding='gbk')
map(lambda x: m_total_data.append(x[0]),
[[line.strip().replace(u"N0", u"NO").split(',')] for line in fhm.readlines()])
fhm.close()
sd = sd + datetime.timedelta(days=1)
from itertools import groupby
c = groupby(sorted(m_total_data, key=lambda s: s[1]), lambda f: f[1])
p_dict = {}
for k, g in c:
p_dict[k] = list(g)
m_sheet = book.add_sheet(prefix + u'日汇总表')
m_headings = [u'店名', u'编号', u'日期', u'交易金额',u'调整金额',u'合计']
m_rowx = 0
for colx, value in enumerate(m_headings):
m_sheet.write(m_rowx, colx, value, styleB)
if len(m_total_data) == 0:
logging.warn(u'生成{prefix}调整金额,error: {error}'.format(prefix=prefix, error=u'数据为空'))
else:
for p_key in p_dict:
c_count = m_rowx+2
for row in p_dict[p_key]:
m_rowx += 1
row.pop(3)
row.pop(4)
for colx, value in enumerate(row):
if colx == 3:
value = float(value)
m_sheet.write(m_rowx, colx, value, styleA)
m_rowx += 1
for colx, value in enumerate(['', p_key+u' 汇总', '', \
Formula("SUBTOTAL(9,D%d:D%d)" % (c_count, m_rowx))]):
if colx == 3:
m_sheet.write(m_rowx, colx, value, styleA)
else:
m_sheet.write(m_rowx, colx, value, styleB)
m_rowx += 1
for colx, value in enumerate(['', u'总计', '', \
Formula("SUBTOTAL(9,D%d:D%d)" % (2, m_rowx-1))]):
if colx == 3:
m_sheet.write(m_rowx, colx, value, styleA)
else:
m_sheet.write(m_rowx, colx, value, styleB)
except Exception, e:
logging.warn(u'生成{prefix}日汇总表,error: {error}'.format(prefix=prefix, error=e))
"""
month to month
"""
try:
fh = codecs.open(filename=check_file_path, mode='r', encoding='gbk')
data1 = map(lambda x: x[0], [[line.strip().split(',')] for line in fh.readlines()])
fh.close()
sheet = book.add_sheet(prefix + u'月明细表')
rowx = 0
for colx, value in enumerate(headings):
sheet.write(rowx, colx, value, styleB)
if len(data1) == 0:
logging.warn(u'生成{prefix}月明细表,error: {error}'.format(prefix=prefix, error=u'数据为空'))
else:
for row in data1:
rowx += 1
for colx, value in enumerate(row):
if colx == 2:
value = float(value)
sheet.write(rowx, colx, value, styleA)
rowx += 1
for colx, value in enumerate(['', u'总计', Formula("SUBTOTAL(9,C%d:C%d)" % (2, rowx))]):
if colx == 2:
sheet.write(rowx, colx, value, styleA)
else:
sheet.write(rowx, colx, value, styleB)
logging.info(u'执行完成，请查看：'+save_file_path)
book.add_sheet(prefix + u'手续费月汇总表')
book.save(save_file_path)
except Exception, e:
logging.error(u'生成{prefix}月明细表,error: {error}'.format(prefix=prefix, error=e))