diff env/lib/python3.9/site-packages/galaxy/util/ucsc.py @ 0:4f3585e2f14b draft default tip

"planemo upload commit 60cee0fc7c0cda8592644e1aad72851dec82c959"

--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/env/lib/python3.9/site-packages/galaxy/util/ucsc.py	Mon Mar 22 18:12:50 2021 +0000
@@ -0,0 +1,39 @@
+"""
+Utilities for dealing with UCSC data.
+"""
+
+
+class UCSCLimitException(Exception):
+    pass
+
+
+class UCSCOutWrapper:
+    """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 next(self)