From b14e3395644c6b8d649a4a81718aad0fafe5b4e3 Mon Sep 17 00:00:00 2001
From: Yuval Adam <yuv.adm@gmail.com>
Date: Mon, 21 Jan 2013 13:04:37 +0200
Subject: Initial regex for ballot extraction

---
 ballots.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)
 create mode 100644 ballots.py

(limited to 'ballots.py')

diff --git a/ballots.py b/ballots.py
new file mode 100644
index 0000000..d822d84
--- /dev/null
+++ b/ballots.py
@@ -0,0 +1,17 @@
+import re
+
+def read_file(filename):
+    with open(filename, 'r') as f:
+        return f.read()
+
+def get_ballots(f):
+    BALLOT_DATA_RE = re.compile(
+        r'(\d\d)\n(.*)\n(\d\d\d\d)\n(.*)\n([\d\.]+)\n(.*)\n(.*)')
+    m = re.findall(BALLOT_DATA_RE, f)
+    for x in m:
+        commitee_id, commitee, city_id, city, ballot_id, ballot_address, ballot_location = x
+        print commitee_id, commitee, city_id, city, ballot_id, ballot_address, ballot_location
+
+f = read_file('ballots.txt')
+data = get_ballots(f)
+print data
-- 
cgit v1.3.1