Switch the location of log/history file based on OS.

3 files changed, 16 insertions, 6 deletions

diff --git a/pgcli/config.py b/pgcli/config.py
index 3911362a..5f46e7e5 100644
--- a/pgcli/config.py
+++ b/pgcli/config.py
@@ -6,9 +6,9 @@ from configobj import ConfigObj
 
 def config_location():
     if platform.system() == 'Windows':
-        return os.getenv('USERPROFILE') + '\AppData\Local\dbcli\pgcli\config'
+        return os.getenv('USERPROFILE') + '\\AppData\\Local\\dbcli\\pgcli\\'
     else:
-        return expanduser('~/.config/pgcli/config')
+        return expanduser('~/.config/pgcli/')
 
 def load_config(usr_cfg, def_cfg=None):
     cfg = ConfigObj()
diff --git a/pgcli/main.py b/pgcli/main.py
index 4cdd3169..5c9e42a1 100755
--- a/pgcli/main.py
+++ b/pgcli/main.py
@@ -154,6 +154,8 @@ class PGCli(object):
     def initialize_logging(self):
 
         log_file = self.config['main']['log_file']
+        if log_file == 'default':
+            log_file = config_location() + 'log'
         log_level = self.config['main']['log_level']
 
         level_map = {'CRITICAL': logging.CRITICAL,
@@ -382,6 +384,8 @@ class PGCli(object):
             ])
 
         history_file = self.config['main']['history_file']
+        if history_file == 'default':
+            history_file = config_location() + 'history'
         with self._completer_lock:
             buf = PGBuffer(
                 always_multiline=self.multi_line,
@@ -535,8 +539,8 @@ class PGCli(object):
 @click.option('-v', '--version', is_flag=True, help='Version of pgcli.')
 @click.option('-d', '--dbname', default='', envvar='PGDATABASE',
         help='database name to connect to.')
-@click.option('--pgclirc', default=config_location(), envvar='PGCLIRC',
-        help='Location of pgclirc file.')
+@click.option('--pgclirc', default=config_location() + 'config',
+        envvar='PGCLIRC', help='Location of pgclirc file.')
 @click.argument('database', default=lambda: None, envvar='PGDATABASE', nargs=1)
 @click.argument('username', default=lambda: None, envvar='PGUSER', nargs=1)
 def cli(database, user, host, port, prompt_passwd, never_prompt, dbname,
diff --git a/pgcli/pgclirc b/pgcli/pgclirc
index 15c39b6b..d592454b 100644
--- a/pgcli/pgclirc
+++ b/pgcli/pgclirc
@@ -16,10 +16,16 @@ wider_completion_menu = False
 multi_line = False
 
 # log_file location.
-log_file = ~/.config/pgcli/log
+# In Unix/Linux: ~/.config/pgcli/log
+# In Windows: %USERPROFILE%\AppData\Local\dbcli\pgcli\log
+# %USERPROFILE% is typically C:\Users\{username}
+log_file = default
 
 # history_file location.
-history_file = ~/.config/pgcli/history
+# In Unix/Linux: ~/.config/pgcli/history
+# In Windows: %USERPROFILE%\AppData\Local\dbcli\pgcli\history
+# %USERPROFILE% is typically C:\Users\{username}
+history_file = default
 
 # Default log level. Possible values: "CRITICAL", "ERROR", "WARNING", "INFO"
 # and "DEBUG".