Only strip off everything but the last character if

the destination ends with a /, otherwise, rip off the whole destination root
hchonan · Jan 8, 2013 · ebcd0c9 · ebcd0c9
1 parent 9b67865
commit ebcd0c9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -45,7 +45,10 @@ def run (self):
 						replace_me = os.path.join(self.install_data,'share/qtarot')
 					elif self.prefix:
 						replace_me = os.path.join(self.prefix,'share/qtarot')
-					consts = [['DATA_DIR', replace_me.replace(self.root[:-1],'')]]
+					if self.root.endswidth('/'):
+						consts = [['DATA_DIR', replace_me.replace(self.root[:-1],'')]]
+					else:
+						consts = [['DATA_DIR', replace_me.replace(self.root,'')]]
 					script = open(f, 'w', encoding='utf-8')
 					script.write(content[:const_begin] + \
 								 "### CONSTANTS BEGIN ###")