Update combineKinship.py

zhanxw · Sep 7, 2017 · 72ab223 · 72ab223
1 parent 1dbbbd9
commit 72ab223
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/misc/combineKinship.py b/misc/combineKinship.py
@@ -1,11 +1,17 @@
 #!/usr/bin/env python
 import sys, os, re
-try:
-    from XiaoweiLib import myopen
-except:
-    sys.path = [re.sub(r'^/home/zhanxw/', '/net/fantasia/home/zhanxw/', x) for x in sys.path]
-    sys.path.append('/net/nfsb/fantasia/home/zhanxw/mylib/Python/')
-    from XiaoweiLib import myopen
+
+# convenient functions
+def myopen(fn):
+    import gzip
+    f = gzip.open(fn)
+    try:
+        f.read(2)
+        f.close()
+        return gzip.open(fn)
+    except:
+        f.close()
+        return open(fn)
 
 def usage():
     print("%s -o prefix in1.kinship in2.kinship ..." % sys.argv[0] )