-
Notifications
You must be signed in to change notification settings - Fork 0
/
position_change.py
78 lines (68 loc) · 2.25 KB
/
position_change.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
76
77
78
#!/usr/bin/python
from optparse import OptionParser
import os
import csv
import sys
import logging
import traceback
import shutil
log = logging.getLogger('python_logger')
log.setLevel(logging.DEBUG)
# create formatter and add it to the handlers
# formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
# 2015-08-28 17:01:57,662 - simple_example - ERROR - error message
# formatter = logging.Formatter('%(asctime)s %(levelname)-8s %(filename)s:%(lineno)-4d: %(message)s')
formatter = logging.Formatter('%(asctime)s %(levelname)-2s [%(filename)s: %(lineno)d] %(message)s')
# create console handler with a higher log level
ch = logging.StreamHandler()
ch.setLevel(logging.DEBUG)
ch.setFormatter(formatter)
# add the handlers to the logger
log.addHandler(ch)
usage = "usage: position.py [options] arg"
parser = OptionParser(usage)
parser.add_option("-d", "--directory", dest="directoryname", help="state file directory", metavar="DIR")
options, args = parser.parse_args()
if options.directoryname is None: # if filename is not given
log.error('Directoryname not given')
sys.exit(1)
directory = options.directoryname
log.info('Directory name: %s', directory)
try:
os.chdir(directory)
file_list = os.listdir('.')
for filename in file_list:
if filename.endswith("_Pos.csv"):
f1 = open(filename, 'r')
f2 = open(filename+'_new', 'w')
reader = csv.reader(f1)
writer = csv.writer(f2)
rownum = 0
for row in reader:
if rownum == 0:
header = row
writer.writerow(row)
else:
l = row
l[1] = int(l[1]) + int(l[3])
l[2] = int(l[2]) + int(l[4])
l[3] = 0
l[4] = 0
if (len(l) >= 8) :
l[7] = 0
writer.writerow(row)
rownum += 1
f1.close()
f2.close()
#log.info("old position :%s", filename)
#os.system("cat " + filename)
#log.info("new position :%s", filename + '_new')
#os.system("cat " + filename +'_new')
shutil.move(filename, filename+'_old')
shutil.move(filename+'_new', filename)
log.info("Finished position change on %s", filename)
os.system('for F in `ls | grep -E "Pos|Holding"`; do echo $F; cat $F; echo ""; done')
except Exception as e:
log.error("Got an exception: %s", traceback.format_exc())
sys.exit(1)
log.info("Finished on: %s", directory)