1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
From: Christian Kastner <ckk@kvr.at>
Date: Tue, 6 Jul 2010 21:26:16 +0200
Subject: [PATCH] Support custom config file location
This patch adds the ability to specify a custom config file via the environment
variable PYRIT_CONFIG_FILE.
Forwarded: yes
Last-Update: 2010-07-06
Index: pyrit-0.4.0/cpyrit/config.py
===================================================================
--- pyrit-0.4.0.orig/cpyrit/config.py 2011-02-14 19:51:21.000000000 +0100
+++ pyrit-0.4.0/cpyrit/config.py 2011-10-09 00:24:01.436006178 +0200
@@ -58,10 +58,18 @@
configpath = os.path.expanduser(os.path.join('~', '.pyrit'))
default_configfile = os.path.join(configpath, 'config')
-if os.path.exists(default_configfile):
- cfg = read_configfile(default_configfile)
+if 'PYRIT_CONFIG_FILE' in os.environ:
+ custom_config = os.environ['PYRIT_CONFIG_FILE']
+ if not os.path.exists(custom_config):
+ print >> sys.stderr, \
+ "WARNING: custom config file %s does not exist" % custom_config
+ else:
+ cfg = read_configfile(custom_config)
else:
- cfg = default_config()
- if not os.path.exists(configpath):
- os.makedirs(configpath)
- write_configfile(cfg, default_configfile)
+ if os.path.exists(default_configfile):
+ cfg = read_configfile(default_configfile)
+ else:
+ cfg = default_config()
+ if not os.path.exists(configpath):
+ os.makedirs(configpath)
+ write_configfile(cfg, default_configfile)
|
