summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--combox/config.py76
1 files changed, 76 insertions, 0 deletions
diff --git a/combox/config.py b/combox/config.py
new file mode 100644
index 0000000..ef83f6d
--- /dev/null
+++ b/combox/config.py
@@ -0,0 +1,76 @@
+# Copyright (C) 2014 Combox author(s). See AUTHORS.
+#
+# This file is part of Combox.
+#
+# Combox is free software: you can redistribute it and/or modify it
+# under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Combox is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with Combox (see COPYING). If not, see
+# <http://www.gnu.org/licenses/>.
+
+##################################################
+## YAML Config format
+##################################################
+##
+## combox_dir: path/to/combox/
+##
+## topsecret: dba0079f1cb3a3b56e102dd5e04fa2af
+##
+## nodes_info:
+## node_name:
+## path: path/to/shard1/dir/
+## size: 1000 # in MB
+## available: 500 # in MB
+## node name:
+## path: path/to/shard2/dir/
+## size: 2000
+## available: 1500
+## node name:
+## path: path/to/shard3/dir
+## size: 3000
+## available: 1500
+##
+##################################################
+
+import os
+import yaml
+import getpass
+import md5
+import sys
+import stat
+
+# First whether combox is already configured.
+config_dir = os.path.join(os.getenv('HOME'),'.combox/')
+if not os.path.exists(config_dir):
+ # Create combox dir and configure.
+ os.mkdir(config_dir, 0700)
+ config_file_path = os.path.join(config_dir, 'config.yaml')
+ config_info = {}
+
+ config_info['combox_dir'] = raw_input('path to combox directory: ')
+ config_info['topsecret'] = md5.new(getpass.getpass('passphrase: ')).hexdigest()
+
+ no_nodes = int(raw_input('number of nodes: '))
+
+ nodes = {}
+ for i in range(no_nodes):
+ node_name = raw_input('node %d name: ' % i)
+ nodes[node_name] = {}
+ nodes[node_name]['path'] = raw_input('node %d path: ' % i)
+ nodes[node_name]['size'] = raw_input('node %d size (in mega bytes): ' % i)
+ nodes[node_name]['available'] = nodes[node_name]['size']
+
+ config_info['nodes_info'] = nodes
+ config_file = open(config_file_path, 'w')
+ yaml.dump(config_info, config_file, default_flow_style=False)
+ os.chmod(config_file_path,stat.S_IRUSR|stat.S_IWUSR)
+else:
+ print "You've already configured combox!"