main.py

#!/usr/bin/env python

__version__ = '0.8'

import sys, os, os.path


def abspath(p, mkdir = False):
	if not p:
		return None
	if p[0] == '~':
		if os.environ['HOME']:
			p = os.environ['HOME'] + p[1:]
		elif os.environ['USERPROFILE']:
			p = os.environ['USERPROFILE'] + p[1:]
		else:
			raise Exception("don't know how to translate ~ to $HOME")
	elif not os.path.isabs(p):
		p = os.path.join(os.path.dirname(__file__), p)
	p = os.path.abspath(p)
	if mkdir and not os.path.exists(p):
		os.makedirs(p)
	return p

def _args():
	import argparse
	p = argparse.ArgumentParser(description = "Kindle Store proxy, v%s" % __version__ , prog = 'ksp')
	p.add_argument('--config', dest = 'etc_path', default = 'etc', help = "Path to the configration folder")
	p.add_argument('--loglevel', dest = 'log_level', choices = ('debug', 'info', 'warn'), default = None,
								help = "Set the minimum logging level in the server, overrides the value given in config.py")
	p.add_argument('--console', dest = 'console', action = 'store_const', const = True, default = False,
								help = "Log to the console instead of a log file")
	p.add_argument('--control-pipe', dest = 'control_pipe', help = "Use the given pipe to read server control commands")
	return p.parse_args()


import logging


def _make_root_logger(stream, log_level = 'NOTSET'):
	print(log_level)
	handler = logging.StreamHandler(stream)
	fmt = logging.Formatter('[%(asctime)-12s.%(msecs)03d] %(levelname)-8s {%(threadName)-10s %(filename)s:%(lineno)d %(funcName)s} %(message)s', '%Y-%m-%d %H:%M:%S')
	handler.setFormatter(fmt)
	log = logging.getLogger()
	log.setLevel(log_level.upper())
	log.addHandler(handler)
	logging.captureWarnings(True)
	return log

def _stdstream(path = None):
	from codecs import open as codecs_open

	if path:
		server_log = os.path.join(path, 'server.log')
		sys.stdout = codecs_open(server_log, mode = 'a', encoding = 'latin1', errors = 'backslashreplace')
		sys.stderr = sys.stdout # open(os.path.join(config.logs_path, 'stderr.log'), 'w', 1)

	if sys.__stdout__ is None:
		sys.__stdout__ = sys.stdout
	if sys.__stderr__ is None:
		sys.__stderr__ = sys.stderr

	return sys.stdout

def _processConfiguration():
	def _check_field(obj, name, defaultValue):
		if not hasattr(obj, name):
			setattr(obj, name, defaultValue)

	import config
	_check_field(config, 'log_level', 'info')
	_check_field(config, 'server_certificate', None)
	_check_field(config, 'disconnected', False)

	import features
	_check_field(features, 'automatic_collections', False)
	_check_field(features, 'collection_tags', [])


def main():
	args = _args()
	etc_path = abspath(args.etc_path)
	sys.path.append(etc_path)

	_processConfiguration()

	import config
	config.logs_path = None if args.console else abspath(config.logs_path, True)
	if args.log_level:
		config.log_level = args.log_level
	_make_root_logger(_stdstream(config.logs_path), config.log_level)

	logging.info("%s start-up", '*' * 20)
	logging.info("read configuration from %s", etc_path)

	config.database_path = abspath(config.database_path, True)
	config.server_certificate = abspath(config.server_certificate)

	# add the src/ folder to the import path
	sys.path.append(abspath('src'))

	# doing this here because if the pipe does not exit, we want to fail fast,
	# before we load the devices and calibre databases
	import ctrl
	pipe_file = open(args.control_pipe, 'rb') if args.control_pipe else None

	try:
		# import calibre and devices here because the config has to be fully processed for them to work
		import devices
		import calibre
		import server
		http_server = server.HTTP()
		if pipe_file:
			ctrl.start_server_controller(http_server, pipe_file, args.control_pipe)
		http_server.run()
	except:
		logging.exception("")
	finally:
		if pipe_file:
			try: pipe_file.close()
			except: pass
		devices.save_all()

	logging.info("%s shutdown", '*' * 20)
	logging.shutdown()


if __name__ == '__main__':
	main()
	os._exit(0)