From afc4138898faf0e901ca08823465679e176fbbe4 Mon Sep 17 00:00:00 2001 From: Binh Nguyen Date: Tue, 19 Feb 2013 16:06:11 -0600 Subject: [PATCH] fixed automatic model generation inheretence --- scripts/pyaccuwage-pdfparse | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/scripts/pyaccuwage-pdfparse b/scripts/pyaccuwage-pdfparse index 3cb781b..6a35387 100755 --- a/scripts/pyaccuwage-pdfparse +++ b/scripts/pyaccuwage-pdfparse @@ -14,17 +14,12 @@ args = parser.parse_args() def generate_imports(): return "\n".join([ - "from pyaccuwage import model", + "from pyaccuwage import model as pyaccuwagemodel", "from pyaccuwage.fields import *", "", "", ]) -def generate_class_begin(name): - name = re.sub(r"^[\d]*", "", name) - print name - return "class %s(mode.Model):\n" % name - if args.full: sys.stdout.write(generate_imports()) @@ -71,7 +66,7 @@ for rec in records: if last_record_ends_at + 1 != begins_at: name = re.sub('^[^a-zA-Z]*','',rec[0].split(':')[-1]) name = re.sub('[^\w]*', '', name) - sys.stdout.write("\nclass %s(object):\n" % name) + sys.stdout.write("\nclass %s(pyaccuwagemodel.Model):\n" % name) for field in builder.load(map(lambda x:x.tuple, rec[1][0:])): sys.stdout.write('\t' + field + '\n')