Adding defaults for project_path & project_name

victorfsf · Feb 14, 2016 · 5ad10ba · 5ad10ba
1 parent 2ceec3f
commit 5ad10ba
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/iprofile/core/utils.py b/iprofile/core/utils.py
@@ -8,8 +8,10 @@
 
 def get_ipython_name(profile_name, config):
     return '{0}_{1}'.format(
-        slugify(config.get('project_name')) or
-        os.path.basename(os.getcwd()), profile_name
+        slugify(
+            config.get('project_name') or os.path.basename(os.getcwd())
+        ),
+        profile_name
     )
 
 

diff --git a/iprofile/models/config.py b/iprofile/models/config.py
@@ -16,6 +16,16 @@ def __init__(self):
                 'project_name': os.path.basename(os.getcwd())
             }
             self.save()
+        else:
+            if not self._config.get('project_path'):
+                self._config.update({
+                    'project_path': 'iprofiles',
+                })
+
+            if not self._config.get('project_name'):
+                self._config.update({
+                    'project_name': os.path.basename(os.getcwd())
+                })
 
     def read(self):
         self._config = yaml.load(open(self.filepath, 'r')) or {}