script to fix old bailey ids

inpho · May 13, 2015 · cfa2607 · cfa2607
1 parent 94c3613
commit cfa2607
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/ob_ids.py b/ob_ids.py
@@ -0,0 +1,16 @@
+from argparse import ArgumentParser
+import numpy as np
+
+from vsm.corpus import *
+
+parser = ArgumentParser()
+parser.add_argument("corpus_path", help="Path to Existing Corpus File")
+args = parser.parse_args()
+
+c = Corpus.load(args.corpus_path)
+pagenos = [int(md['file'].split('/')[-1].split('_')[0])
+    for md in c.view_metadata('page')]
+pagenos = np.array(pagenos)
+
+c = add_metadata(c, 'page', 'trial_id', pagenos)
+c.save(args.corpus_path)