From a2d45b9c5593e4ecf37850db69d3bcd8c673419d Mon Sep 17 00:00:00 2001
From: Lei Yan
Date: Sat, 29 Oct 2016 05:25:37 +0000
Subject: For those case that we have multiple genotypes we would like to add a
 selector.

---
 wqflask/base/data_set.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'wqflask/base')

diff --git a/wqflask/base/data_set.py b/wqflask/base/data_set.py
index fddfce58..918932fa 100644
--- a/wqflask/base/data_set.py
+++ b/wqflask/base/data_set.py
@@ -278,6 +278,7 @@ class DatasetGroup(object):
         self.incparentsf1 = False
         self.allsamples = None
         self._datasets = None
+        self.genofile = None
 
     def get_specified_markers(self, markers = []):
         self.markers = HumanMarkers(self.name, markers)
@@ -408,7 +409,10 @@ class DatasetGroup(object):
         genotype_1 = reaper.Dataset()
 
         # reaper barfs on unicode filenames, so here we ensure it's a string
-        full_filename = str(locate(self.name+'.geno','genotype'))
+        if self.genofile:
+            full_filename = str(locate(self.genofile, 'genotype'))
+        else:
+            full_filename = str(locate(self.name + '.geno', 'genotype'))
         genotype_1.read(full_filename)
 
         if genotype_1.type == "group" and self.parlist:
-- 
cgit v1.2.3