1919
2020Author: Brett G. Olivier
212122- Last edit: $Author: bgoli $ ($Id: CBModel.py 625 2017-10-20 12:00:53Z bgoli $)
22+ Last edit: $Author: bgoli $ ($Id: CBModel.py 629 2017-10-24 22:01:14Z bgoli $)
2323
2424"""
2525## gets rid of "invalid variable name" info
3434from __future__ import absolute_import
3535#from __future__ import unicode_literals
3636
37- import numpy , re , time , weakref , copy , json , urllib2 , ast
37+ import numpy , re , time , weakref , copy , json , ast
38+
39+ try :
40+ from urllib2 import quote as urlquote
41+ from urllib2 import unquote as urlunquote
42+ except ImportError :
43+ from urllib .parse import quote as urlquote
44+ from urllib .parse import unquote as urlunquote
3845
3946global GENE_CNTR
4047GENE_CNTR = 0
@@ -218,7 +225,10 @@ def setNotes(self, notes):
218225
219226 """
220227 #self.notes = self.__urlEncode(notes)
221- self .notes = notes .decode (errors = 'ignore' )
228+ try :
229+ self .notes = notes .decode (errors = 'ignore' )
230+ except AttributeError :
231+ self .notes = notes
222232
223233 def setAnnotation (self , key , value ):
224234 """
@@ -347,7 +357,7 @@ def serializeToDisk(self, filename, protocol=2):
347357 # Reimplemented in Model
348358
349359 """
350- F = file (filename , 'wb' )
360+ F = open (filename , 'wb' )
351361 pickle .dump (self , F , protocol = protocol )
352362 F .close ()
353363
@@ -440,7 +450,7 @@ def __urlEncode(self, txt):
440450
441451 """
442452 try :
443- txt = urllib2 . quote (txt .encode (self .__text_encoding__ , errors = 'replace' ), safe = '' )
453+ txt = urlquote (txt .encode (self .__text_encoding__ , errors = 'replace' ), safe = '' )
444454 except UnicodeDecodeError as why :
445455 pass
446456 #print(txt)
@@ -451,7 +461,7 @@ def __urlDecode(self, txt):
451461 Decodes a URL encoded string
452462
453463 """
454- return urllib2 . unquote (txt )
464+ return urlunquote (txt )
455465
456466class Model (Fbase ):
457467 """
@@ -716,7 +726,7 @@ def serializeToDisk(self, filename, protocol=2):
716726 # overloaded in CBModel
717727
718728 """
719- F = file (filename , 'wb' )
729+ F = open (filename , 'wb' )
720730 pickle .dump (self , F , protocol = protocol )
721731 F .close ()
722732 self .__setGlobalIdStore__ ()
@@ -4584,7 +4594,7 @@ def addAssociation(self, assoc):
45844594
45854595 """
45864596 #self.assoc = assoc
4587- raise RuntimeError , '\n This method has ceased to exist'
4597+ raise RuntimeError ( '\n This method has ceased to exist' )
45884598
45894599 def createAssociationAndGeneRefsFromTree (self , gprtree , altlabels = None ):
45904600 """
@@ -4633,7 +4643,7 @@ def createAssociationAndGeneRefsFromTree(self, gprtree, altlabels=None):
46334643 self .buildEvalFunc ()
46344644
46354645 def createAssociationAndGeneRefs (self ):
4636- raise RuntimeError , "\n \n DEPRECATED CHANGE NOW!"
4646+ raise RuntimeError ( "\n \n DEPRECATED CHANGE NOW!" )
46374647
46384648 def createAssociationAndGeneRefsFromString (self , assoc , altlabels = None ):
46394649 """
@@ -4688,7 +4698,7 @@ def createAssociationAndGeneRefsFromString(self, assoc, altlabels=None):
46884698 except SyntaxError :
46894699 err = '\n Error in Gene Association String: {}\n ' .format (assoc )
46904700 #print(err)
4691- raise SyntaxError , err
4701+ raise SyntaxError ( err )
46924702 self .setTree (newtree )
46934703 genes = self .__getGeneRefsfromGPRDict__ (newtree , [])
46944704 genes .sort ()
@@ -4788,7 +4798,7 @@ def getAssociationStr(self, use_labels=False):
47884798 keymap = {}
47894799 for g in self .generefs :
47904800 keymap [g ] = self .__objref__ ().getGene (g ).getLabel ()
4791- keys = keymap .keys ()
4801+ keys = list ( keymap .keys () )
47924802 keys .sort (reverse = True )
47934803 for k in keys :
47944804 out = out .replace (k , keymap [k ])
0 commit comments