install.py fixes

1 files changed, 20 insertions, 21 deletions

diff --git a/install.py b/install.py
index 54a8aa3..2a52159 100755
--- a/install.py
+++ b/install.py
@@ -6,26 +6,26 @@ import requests
 from string import Template
 from subprocess import call
 
-from servo.models import Location, User
-
 default_hostname = socket.gethostname()
 tpl_url = 'https://gist.githubusercontent.com/filipp/cba2ffecd0d5790f7245/raw/'
 
-print("** Creating local configuration file **")
+fh = open('local_settings.py', 'w')
 
+print("** Creating local configuration file **")
 args = {}
 args['secret_key'] = os.urandom(32).encode('base-64').strip()
-args['install_country'] = raw_input('Country [SE]: ') or 'SE'
-args['install_locale'] = raw_input('Locale [sv_SE.UTF-8]: ') or 'sv_SE.UTF-8'
-args['install_language'] = raw_input('Language [sv]: ') or 'sv'
-args['timezone'] = raw_input('Timezone [Europe/Stockholm]: ') or 'Europe/Stockholm'
-args['install_id'] = raw_input('Installation ID [22]: ') or '22'
-
-args['hostname'] = raw_input('Hostname [%s]' % default_hostname) or default_hostname
-args['dbhost']   = raw_input('Database host [localhost]: ') or 'localhost'
-args['dbname']   = raw_input('Database [servo]: ') or 'servo'
-args['dbuser']   = raw_input('DB user [servo]: ') or 'servo'
-args['dbpwd']    = raw_input('DB password []: ') or ''
+args['install_locale'] = raw_input('1/10 Locale [sv_SE.UTF-8]: ') or 'sv_SE.UTF-8'
+default_country = args['install_locale'].split('_')[0]
+args['install_country'] = raw_input('2/10 Country [%s]: ' % default_country) or default_country
+default_lang = args['install_locale'].split('_')[1][:2]
+args['install_language'] = raw_input('3/10 Language [%s]: ' % default_lang) or default_lang
+args['timezone'] = raw_input('4/10 Timezone [Europe/Stockholm]: ') or 'Europe/Stockholm'
+args['install_id'] = raw_input('5/10 Installation ID [22]: ') or '22'
+args['hostname'] = raw_input('6/10 Hostname [%s]: ' % default_hostname) or default_hostname
+args['dbhost']   = raw_input('7/10 Database host [localhost]: ') or 'localhost'
+args['dbname']   = raw_input('8/10 Database [servo]: ') or 'servo'
+args['dbuser']   = raw_input('9/10 DB user [servo]: ') or 'servo'
+args['dbpwd']    = raw_input('10/10 DB password []: ') or ''
 
 raw = requests.get(tpl_url).text
 template = Template(raw)
@@ -35,7 +35,6 @@ s = template.substitute(**args)
 call(['createuser', args['dbuser'], '-U', 'pgsql'])
 call(['createdb', args['dbname'], '-O', args['dbuser'], '-U', 'pgsql'])
 
-fh = open('local_settings.py', 'w')
 fh.write(s)
 fh.close()
 
@@ -49,12 +48,12 @@ print("** Creating Super User **")
 call(['./manage.py', 'createsuperuser'])
 
 print("** Creating initial Service Location **")
-loc['title']    = raw_input('Name [PretendCo Inc]: ') or 'PretendCo Inc'
-loc['email']    = raw_input('Email [service@pretendo.com]: ') or 'service@pretendo.com'
-loc['phone']    = raw_input('Phone [123456789]: ') or '123456789'
-loc['address']  = raw_input('Address [Somestreet 1]: ') or 'Somestreet 1'
-loc['zip_code'] = raw_input('Postal code [1234]: ') or '1234'
-loc['city']     = raw_input('City [Stockholm]: ') or 'Stockholm'
+loc['title']    = raw_input('1/6 Name [PretendCo Inc]: ') or 'PretendCo Inc'
+loc['email']    = raw_input('2/6 Email [service@pretendo.com]: ') or 'service@pretendo.com'
+loc['phone']    = raw_input('3/6 Phone [123456789]: ') or '123456789'
+loc['address']  = raw_input('4/6 Address [Somestreet 1]: ') or 'Somestreet 1'
+loc['zip_code'] = raw_input('5/6 Postal code [1234]: ') or '1234'
+loc['city']     = raw_input('6/6 City [Stockholm]: ') or 'Stockholm'
 
 first_loc = Location(**loc)
 first_loc.save()