diff planemo/lib/python3.7/site-packages/galaxy/util/ucsc.py @ 1:56ad4e20f292 draft

"planemo upload commit 6eee67778febed82ddd413c3ca40b3183a3898f1"
author guerler
date Fri, 31 Jul 2020 00:32:28 -0400
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/planemo/lib/python3.7/site-packages/galaxy/util/ucsc.py	Fri Jul 31 00:32:28 2020 -0400
@@ -0,0 +1,39 @@
+"""
+Utilities for dealing with UCSC data.
+"""
+
+
+class UCSCLimitException(Exception):
+    pass
+
+
+class UCSCOutWrapper(object):
+    """File-like object that throws an exception if it encounters the UCSC limit error lines"""
+
+    def __init__(self, other):
+        self.other = iter(other)
+        # Need one line of lookahead to be sure we are hitting the limit message
+        self.lookahead = None
+
+    def __iter__(self):
+        return self
+
+    def __next__(self):
+        if self.lookahead is None:
+            line = next(self.other)
+        else:
+            line = self.lookahead
+            self.lookahead = None
+        if line.startswith("----------"):
+            next_line = next(self.other)
+            if next_line.startswith("Reached output limit"):
+                raise UCSCLimitException(next_line.strip())
+            else:
+                self.lookahead = next_line
+        return line
+
+    def next(self):
+        return self.__next__()
+
+    def readline(self):
+        return self.next()