zoo.parse.parse_location more generic

phiweger · Mar 22, 2017 · fd81567 · fd81567
1 parent 58af441
commit fd81567
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/zoo/parse.py b/zoo/parse.py
@@ -72,3 +72,19 @@ def parse_date(date):
         except (IndexError, ValueError):
             continue
     return record
+
+
+def parse_location(location, source='genbank'):
+    '''
+    Example:
+
+    parse_gb_location('1->1374')
+    # ('1', '1374', 1)  # start, end, fuzzy
+    '''
+    if source == 'genbank':
+        start, end = re.findall('\\d+', location)
+        fuzzy = int(any([i in location for i in ['>', '<']]))
+        return start, end, fuzzy
+    raise AttributeError('Unknown format.')
+
+