pythonrc.symlink

# vim:filetype=python
print("Running pythonrc")

import datetime

try:
    get_ipython().confirm_exit = False
except:
    print('Not running in ipython')

try:
    pytz = __import__('pytz')
except ImportError:
    print("Can't find pytz - run: pip install pytz")

try:
    dateutil = __import__('dateutil')
except ImportError:
    print("Can't find dateutil - run: pip install python-dateutil")
else:
    from dateutil.relativedelta import relativedelta

import atexit
import os
import readline
import rlcompleter
import readline
import rlcompleter

readline.parse_and_bind('tab: complete')
# Don't trust expanduser with just ~ - it defaults to the pre-sudo username
user = os.environ['USER']
historyPath = os.path.expanduser('~{}/.pyhistory'.format(user))

def save_history(historyPath=historyPath):
    import readline
    readline.write_history_file(historyPath)

if os.path.exists(historyPath):
    readline.read_history_file(historyPath)

atexit.register(save_history)

import csv
def save_csv(filename, data):
  with open(filename, 'w', newline='') as csvfile:
      writer = csv.DictWriter(csvfile, [type(list(data.keys())[0]), 'value'])
      for key, value in sorted(data.items()):
        writer.writerow({type(key): key, 'value': value})

import logging
log = logging.getLogger(__name__)

del os, atexit, readline, rlcompleter, save_history, historyPath