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
|
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
# AWL simulator - Symbol table parser
#
# Copyright 2014-2016 Michael Buesch <m@bues.ch>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
#
from __future__ import division, absolute_import, print_function, unicode_literals
import sys
import getopt
from awlsim_loader.common import *
from awlsim_loader.core import *
def usage():
print("awlsim-symtab symbol table parser, version %s" %\
VERSION_STRING)
print("")
print("Usage: awlsim-symtab [OPTIONS] [inputfile] [outputfile]")
print("")
print("If inputfile is - or omitted, stdin is used.")
print("If outputfile is - or omitted, stdout is used.")
print("")
print("Options:")
print(" -I|--input-format FMT Input file format.")
print(" FMT may be one of: auto, csv, asc")
print(" Default: auto")
print(" -O|--output-format FMT Input file format.")
print(" FMT may be one of: csv, readable-csv, asc")
print(" Default: readable-csv")
print("")
print("Example usage for converting .ASC to readable .CSV:")
print(" awlsim-symtab -I asc -O readable-csv symbols.asc symbols.csv")
def main():
opt_inputParser = None
opt_outputFormat = "readable-csv"
opt_infile = "-"
opt_outfile = "-"
try:
(opts, args) = getopt.getopt(sys.argv[1:],
"hI:O:",
[ "help", "input-format=", "output-format=", ])
except getopt.GetoptError as e:
printError(str(e))
usage()
return ExitCodes.EXIT_ERR_CMDLINE
for (o, v) in opts:
if o in ("-h", "--help"):
usage()
return ExitCodes.EXIT_OK
if o in ("-I", "--input-format"):
if v.lower() == "auto":
opt_inputParser = None
elif v.lower() in ("csv", "readable-csv"):
opt_inputParser = SymTabParser_CSV
elif v.lower() == "asc":
opt_inputParser = SymTabParser_ASC
else:
printError("Invalid --input-format")
return ExitCodes.EXIT_ERR_CMDLINE
if o in ("-O", "--output-format"):
opt_outputFormat = v.lower()
if opt_outputFormat not in ("csv", "readable-csv", "asc"):
printError("Invalid --output-format")
return ExitCodes.EXIT_ERR_CMDLINE
if len(args) == 1:
opt_infile = args[0]
elif len(args) == 2:
opt_infile = args[0]
opt_outfile = args[1]
elif len(args) > 2:
usage()
return ExitCodes.EXIT_ERR_CMDLINE
try:
if opt_infile == "-":
if isMicroPython:
inDataBytes = sys.stdin.read().encode(SymTabSource.COMPAT_ENCODING)
elif isPy2Compat:
inDataBytes = sys.stdin.read()
else:
inDataBytes = sys.stdin.buffer.read()
else:
inDataBytes = safeFileRead(opt_infile)
# Decode the symbol table using S7 compatible encoding.
encoding = SymTabSource.COMPAT_ENCODING
try:
inDataText = inDataBytes.decode(encoding)
except UnicodeError as e:
raise AwlSimError("Failed to %s decode symbol table text: "
"%s" % (encoding, str(e)))
if opt_inputParser:
tab = opt_inputParser.parseText(inDataText,
autodetectFormat=False)
else:
tab = SymTabParser.parseText(inDataText,
autodetectFormat=True)
if opt_outputFormat == "csv":
outDataText = tab.toCSV()
elif opt_outputFormat == "readable-csv":
outDataText = tab.toReadableCSV()
elif opt_outputFormat == "asc":
outDataText = tab.toASC(stripWhitespace=False)
else:
assert(0)
# Encode the symbol table using S7 compatible encoding.
encoding = SymTabSource.COMPAT_ENCODING
try:
outDataBytes = outDataText.encode(encoding)
except UnicodeError as e:
raise AwlSimError("Failed to %s encode symbol table text: "
"%s" % (encoding, str(e)))
if opt_outfile == "-":
if isMicroPython:
sys.stdout.write(outDataBytes.decode(SymTabSource.COMPAT_ENCODING))
elif isPy2Compat:
sys.stdout.write(outDataBytes)
sys.stdout.flush()
else:
sys.stdout.buffer.write(outDataBytes)
sys.stdout.buffer.flush()
else:
try:
fd = open(opt_outfile, "wb")
fd.write(outDataBytes)
fd.close()
except IOError as e:
printError("Failed to write output file '%s': %s" %\
(opt_outfile, str(e)))
return ExitCodes.EXIT_ERR_IO
except AwlSimError as e:
printError(e.getReport())
return ExitCodes.EXIT_ERR_SIM
return ExitCodes.EXIT_OK
if __name__ == "__main__":
sys.exit(main())
|
