pep8 style

Author: whalebot-helmsman

webstruct/annotation_converter.py

@@ -6,18 +6,21 @@
 
 
 def main():
-    cmdline = argparse.ArgumentParser(description=('utility to convert annotations'
-                                                   'from GATE format to WebAnnotator format'))
-    cmdline.add_argument('--input',
-                         help='path to source annotated file',
+    cmdline = argparse.ArgumentParser(description=('utility '
+                                                   'to convert annotations '
+                                                   'from GATE format to '
+                                                   'WebAnnotator format'))
+    cmdline.add_argument('--GATE',
+                         help='path to file annotated in GATE format',
                          type=str,
                          required=True)
     cmdline.add_argument('--sample',
-                         help='path to already marked html',
+                         help=('path to file annotated in WebAnnotator format '
+                               'for colors and entities transfer'),
                          type=str,
                          required=True)
-    cmdline.add_argument('--output',
-                         help='path to result annotated file',
+    cmdline.add_argument('--WebAnnotator',
+                         help='path to result file in WebAnnotator format',
                          type=str,
                          required=True)
     cmdline.add_argument('--loglevel',
@@ -38,12 +41,12 @@ def main():
 
     gate = webstruct.loaders.GateLoader(known_entities=entities)
     tokenizer = webstruct.HtmlTokenizer(tagset=entities)
-    with open(args.input, 'rb') as reader:
+    with open(args.GATE, 'rb') as reader:
         data = reader.read()
         tree = gate.loadbytes(data)
         tokens, annotations = tokenizer.tokenize_single(tree)
         tree = to_webannotator(tree, entity_colors=colors)
-        with open(args.output, 'wb') as writer:
+        with open(args.WebAnnotator, 'wb') as writer:
             tree.write(writer, method='html', pretty_print=True)
 
 if __name__ == "__main__":