Fabric is now configurable.

Sergio Oliveira
1 parent 4b2544b2
Showing 2 changed files with 18 additions and 20 deletions Show diff stats
.gitignore
fabfile.py
@@ -4,6 +4,7 @@
 *.pyc
 *.pyo
 local_settings.py
+project_cfg.py
  
 *.swp
 *.sqlite3
@@ -3,17 +3,15 @@
 import os
  
 from fabric import colors
+from fabric.utils import error
 from fabric.decorators import task
 from fabric.api import env, run, sudo, local
 from fabric.contrib.files import exists
 from fabric.context_managers import prefix, cd, settings, shell_env
  
  
-### Start of config
-
 APP_USER = APP_NAME = VENV_NAME = 'colab'
-REPO_URL = 'To be defined'
-
+REPO_URL = 'git@github.com:colab-community/colab.git'
  
 environments = {
     'dev': {
@@ -23,23 +21,9 @@ environments = {
         'is_vagrant': True,
         'superuser': 'vagrant',
     },
-    'qa': {
-        'hosts': [],
-        'port': 22,
-        'is_vagrant': False,
-        'superuser': 'root',
-    },
-    'prod': {
-        'hosts': [],
-        'port': 22,
-        'is_vagrant': False,
-        'superuser': 'root',
-    },
 }
 DEFAULT_ENVIRONMENT = 'dev'
  
-### End of config
-
 env.user = APP_USER
 env.use_shell = False
  
@@ -52,11 +36,24 @@ SETTINGS_PATH = os.path.join(MANAGE_PATH, APP_NAME)
  
  
 @task
-def environment(name):
+def environment(name=DEFAULT_ENVIRONMENT):
     """Set the environment where the tasks will be executed"""
+    global REPO_URL
+
+    try:
+        import project_cfg
+    except ImportError:
+        pass
+    else:
+        REPO_URL = project_cfg.repository_url
+        environments.update(project_cfg.environments)
+
+    if name not in environments:
+        error(colors.red('Environment `{}` does not exist.'.format(name)))
+
     env.update(environments[name])
     env.environment = name
-environment(DEFAULT_ENVIRONMENT)
+environment()
  
  
 def aptget_install(pkg):
...	...	@@ -4,6 +4,7 @@
4	4	*.pyc
5	5	*.pyo
6	6	local_settings.py
	7	+project_cfg.py
7	8
8	9	*.swp
9	10	*.sqlite3
...	...
...	...	@@ -3,17 +3,15 @@
3	3	import os
4	4
5	5	from fabric import colors
	6	+from fabric.utils import error
6	7	from fabric.decorators import task
7	8	from fabric.api import env, run, sudo, local
8	9	from fabric.contrib.files import exists
9	10	from fabric.context_managers import prefix, cd, settings, shell_env
10	11
11	12
12		-### Start of config
13		-
14	13	APP_USER = APP_NAME = VENV_NAME = 'colab'
15		-REPO_URL = 'To be defined'
16		-
	14	+REPO_URL = 'git@github.com:colab-community/colab.git'
17	15
18	16	environments = {
19	17	'dev': {
...	...	@@ -23,23 +21,9 @@ environments = {
23	21	'is_vagrant': True,
24	22	'superuser': 'vagrant',
25	23	},
26		- 'qa': {
27		- 'hosts': [],
28		- 'port': 22,
29		- 'is_vagrant': False,
30		- 'superuser': 'root',
31		- },
32		- 'prod': {
33		- 'hosts': [],
34		- 'port': 22,
35		- 'is_vagrant': False,
36		- 'superuser': 'root',
37		- },
38	24	}
39	25	DEFAULT_ENVIRONMENT = 'dev'
40	26
41		-### End of config
42		-
43	27	env.user = APP_USER
44	28	env.use_shell = False
45	29
...	...	@@ -52,11 +36,24 @@ SETTINGS_PATH = os.path.join(MANAGE_PATH, APP_NAME)
52	36
53	37
54	38	@task
55		-def environment(name):
	39	+def environment(name=DEFAULT_ENVIRONMENT):
56	40	"""Set the environment where the tasks will be executed"""
	41	+ global REPO_URL
	42	+
	43	+ try:
	44	+ import project_cfg
	45	+ except ImportError:
	46	+ pass
	47	+ else:
	48	+ REPO_URL = project_cfg.repository_url
	49	+ environments.update(project_cfg.environments)
	50	+
	51	+ if name not in environments:
	52	+ error(colors.red('Environment `{}` does not exist.'.format(name)))
	53	+
57	54	env.update(environments[name])
58	55	env.environment = name
59		-environment(DEFAULT_ENVIRONMENT)
	56	+environment()
60	57
61	58
62	59	def aptget_install(pkg):
...	...