/
merge.py
64 lines (49 loc) · 1.38 KB
/
merge.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
#coding: UTF-8
#
# merge.py
#
# usage: merge.py [dir]
# ex) merge.py
# ex) merge.py D:\testdir
#
import sys
import os
import csv
import datetime
# main
if __name__ == '__main__':
args = sys.argv
if len(args) < 1:
print("引数を確認してください:" + args[0])
sys.exit(1)
# ディレクトリの確認
topdir = os.getcwd()
date = str(datetime.date.today())
if len(args) == 2:
savedir = args[1] + '\\'
else:
savedir = topdir + '\\' + date +'\\'
tmpdir = savedir + 'tmp\\'
if not os.path.exists(savedir):
print("対象ディレクトリがありません:" + savedir)
sys.exit(1)
if not os.path.exists(tmpdir):
print("作業ディレクトリがありません:" + tmpdir)
sys.exit(1)
##### ファイル結合
files = os.listdir(tmpdir)
if len(files) == 0:
print("作業ファイルがありません")
sys.exit(1)
fm = open(tmpdir + 'merge.txt', 'w')
for file in files:
f = open(tmpdir + file, 'r')
fm.write(f.read())
fm.write('\n')
f.close()
fm.close()
save_csv = savedir + date + '.csv'
if os.path.isfile(save_csv):
filenum = len(os.listdir(savedir))
os.rename(save_csv, save_csv + '_' + str(filenum - 1))
os.rename(tmpdir + 'merge.txt', save_csv)