75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
from ConfigParser import ConfigParser
|
|
import os
|
|
import easygui
|
|
import platform
|
|
|
|
def get_config_dir(make=False):
|
|
if platform.system() == 'Windows':
|
|
config_suffix = os.path.join('AppData', 'Local', 'ed_tools')
|
|
else:
|
|
config_suffix = '.ed_tools'
|
|
|
|
return os.path.join(os.path.expanduser('~'), config_suffix)
|
|
|
|
def get_settings(use_gui=True):
|
|
"""
|
|
Try to read the settings from file into ConfigParser object.
|
|
If the config file isn't found, initialize it.
|
|
"""
|
|
filename = os.path.join(get_config_dir(), 'settings.conf')
|
|
settings = ConfigParser()
|
|
|
|
if os.path.isfile(filename):
|
|
settings.read(filename)
|
|
else:
|
|
try:
|
|
os.makedirs(get_config_dir())
|
|
except:
|
|
pass
|
|
|
|
settings = init_settings(use_gui)
|
|
|
|
return settings
|
|
|
|
def init_settings(gui=True):
|
|
settings = ConfigParser()
|
|
settings.add_section('ed_companion')
|
|
settings.add_section('inara')
|
|
if gui:
|
|
_settings_prompt_gui(settings)
|
|
else:
|
|
_settings_prompt_cli(settings)
|
|
print "To change these settings later, edit " + filename
|
|
|
|
with open(os.path.join(get_config_dir(), 'settings.conf'), 'wb') as f:
|
|
settings.write(f)
|
|
|
|
return settings
|
|
|
|
def _settings_prompt_gui(settings):
|
|
data = []
|
|
data = easygui.multenterbox(
|
|
"Enter your E:D and Inara credentials. You only need to do this once.",
|
|
"Authentication Data",
|
|
["Elite Username (email address)", "Elite Password",
|
|
"Inara Username", "Inara Password"]
|
|
)
|
|
|
|
for i in range(4):
|
|
if data[i].strip() == '':
|
|
easygui.msgbox("You must provide data for all fields.")
|
|
_settings_prompt_gui(settings)
|
|
return
|
|
|
|
settings.set('ed_companion', 'username', data[0].strip())
|
|
settings.set('ed_companion', 'password', data[1].strip())
|
|
settings.set('inara', 'username', data[2].strip())
|
|
settings.set('inara', 'password', data[3].strip())
|
|
|
|
|
|
def _settings_prompt_cli(settings):
|
|
settings.set('ed_companion', 'username', raw_input("Elite Username (email address): "))
|
|
settings.set('ed_companion', 'password', raw_input("Elite Password: "))
|
|
settings.set('inara', 'username', raw_input("Inara Username: "))
|
|
settings.set('inara', 'password', raw_input("Inara Password: "))
|