added --datadir argument

p2pool · Feb 9, 2012 · ff2a677 · ff2a677
1 parent 1f343ea
commit ff2a677
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/p2pool/main.py b/p2pool/main.py
@@ -963,6 +963,9 @@ def convert_arg_line_to_args(self, arg_line):
     parser.add_argument('-a', '--address',
         help='generate payouts to this address (default: <address requested from bitcoind>)',
         type=str, action='store', default=None, dest='address')
+    parser.add_argument('--datadir',
+        help='store data in this directory (default: <directory run_p2pool.py is in>/data)',
+        type=str, action='store', default=None, dest='datadir')
     parser.add_argument('--logfile',
         help='''log to this file (default: data/<NET>/log)''',
         type=str, action='store', default=None, dest='logfile')
@@ -1024,7 +1027,7 @@ def convert_arg_line_to_args(self, arg_line):
     net_name = args.net_name + ('_testnet' if args.testnet else '')
     net = networks.nets[net_name]
 
-    datadir_path = os.path.join(os.path.dirname(sys.argv[0]), 'data', net_name)
+    datadir_path = os.path.join((os.path.join(os.path.dirname(sys.argv[0]), 'data') if args.datadir is None else args.datadir), net_name)
     if not os.path.exists(datadir_path):
         os.makedirs(datadir_path)