mirrors/core-lightning
1
0
Fork 0
mirror of https://github.com/ElementsProject/lightning.git synced 2024-11-19 09:54:16 +01:00
Code Issues Projects Releases Wiki Activity

tools/generate-wire.py: Apparently OptionParser is deprecated, use argparse.

Browse Source 
It is a bit nicer though.

Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
This commit is contained in:
Rusty Russell 2017-01-06 13:24:45 +10:30
parent 8b72604d4d
commit 2057de6d4f
1 changed files with 19 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
tools/generate-wire.py
View File

@ -1,7 +1,7 @@
#! /usr/bin/python3 #! /usr/bin/python3
# Read from stdin, spit out C header or body. # Read from stdin, spit out C header or body.
from optparse import OptionParser import argparse
from collections import namedtuple from collections import namedtuple
import fileinput import fileinput
import re import re
@ -322,18 +322,15 @@ class Message(object):
'\treturn memcheck(p, tal_count(p));\n' '\treturn memcheck(p, tal_count(p));\n'
'}\n') '}\n')
parser = OptionParser() parser = argparse.ArgumentParser(description='Generate C from from CSV')
parser.add_option("--header", parser.add_argument('--header', action='store_true', help="Create wire header")
action="store_true", dest="output_header", default=False, parser.add_argument('headerfilename', help='The filename of the header')
help="Create wire header") parser.add_argument('enumname', help='The name of the enum to produce')
parser.add_argument('files', nargs='*', help='Files to read in (or stdin)')
options = parser.parse_args()
(options, args) = parser.parse_args() if options.header:
idem = re.sub(r'[^A-Z]+', '_', options.headerfilename.upper())
if len(args) != 2:
parser.error("Expect headerfilename and enumname")
if options.output_header:
idem = re.sub(r'[^A-Z]+', '_', args[0].upper())
print('#ifndef LIGHTNING_{0}\n' print('#ifndef LIGHTNING_{0}\n'
'#define LIGHTNING_{0}\n' '#define LIGHTNING_{0}\n'
'#include <ccan/tal/tal.h>\n' '#include <ccan/tal/tal.h>\n'
@ -342,7 +339,7 @@ else:
print('#include <{}>\n' print('#include <{}>\n'
'#include <ccan/mem/mem.h>\n' '#include <ccan/mem/mem.h>\n'
'#include <ccan/tal/str/str.h>\n' '#include <ccan/tal/str/str.h>\n'
''.format(args[0])) ''.format(options.headerfilename))
# Maps message names to messages # Maps message names to messages
messages = [] messages = []
@ -350,7 +347,7 @@ comments = []
includes = [] includes = []
# Read csv lines. Single comma is the message values, more is offset/len. # Read csv lines. Single comma is the message values, more is offset/len.
for line in fileinput.input(args[2:]): for line in fileinput.input(options.files):
# #include gets inserted into header # #include gets inserted into header
if line.startswith('#include '): if line.startswith('#include '):
includes.append(line) includes.append(line)
@ -383,24 +380,24 @@ for line in fileinput.input(args[2:]):
break break
comments=[] comments=[]
if options.output_header: if options.header:
for i in includes: for i in includes:
print(i, end='') print(i, end='')
print('') print('')
# Dump out enum, sorted by value order. # Dump out enum, sorted by value order.
print('enum {} {{'.format(args[1])) print('enum {} {{'.format(options.enumname))
for m in messages: for m in messages:
for c in m.comments: for c in m.comments:
print('\t/*{} */'.format(c)) print('\t/*{} */'.format(c))
print('\t{} = {},'.format(m.enum.name, m.enum.value)) print('\t{} = {},'.format(m.enum.name, m.enum.value))
print('};') print('};')
print('const char *{}_name(int e);'.format(args[1])) print('const char *{}_name(int e);'.format(options.enumname))
else: else:
print('const char *{}_name(int e)'.format(args[1])) print('const char *{}_name(int e)'.format(options.enumname))
print('{{\n' print('{{\n'
'\tswitch ((enum {})e) {{'.format(args[1])); '\tswitch ((enum {})e) {{'.format(options.enumname));
for m in messages: for m in messages:
print('\tcase {0}: return "{0}";'.format(m.enum.name)) print('\tcase {0}: return "{0}";'.format(m.enum.name))
print('\t}\n' print('\t}\n'
@ -409,10 +406,10 @@ else:
'') '')
for m in messages: for m in messages:
m.print_fromwire(options.output_header) m.print_fromwire(options.header)
for m in messages: for m in messages:
m.print_towire(options.output_header) m.print_towire(options.header)
if options.output_header: if options.header:
print('#endif /* LIGHTNING_{} */\n'.format(idem)) print('#endif /* LIGHTNING_{} */\n'.format(idem))