version-dependent str type

Signed-off-by: Chris Warrick <kwpolska@gmail.com>
getnikola · May 21, 2015 · 5c59b19 · 5c59b19
1 parent da075a0
commit 5c59b19
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/nikola/utils.py b/nikola/utils.py
@@ -1148,8 +1148,13 @@ def get_root_dir():
     """Find root directory of nikola installation by looking for conf.py"""
     root = os.getcwd()
 
+    if sys.version_info[0] == 3:
+        confname = b'conf.py'
+    else:
+        confname = 'conf.py'
+
     while True:
-        if os.path.exists(os.path.join(root, b'conf.py')):
+        if os.path.exists(os.path.join(root, confname)):
             return root
         else:
             basedir = os.path.split(root)[0]