71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import asyncio
|
|
|
|
import uvloop
|
|
import sys
|
|
|
|
from logbook import Logger, StreamHandler
|
|
|
|
from wallabag_kindle_consumer.config import Config
|
|
from wallabag_kindle_consumer.consumer import Consumer
|
|
from wallabag_kindle_consumer.interface import App
|
|
from wallabag_kindle_consumer.refresher import Refresher
|
|
from wallabag_kindle_consumer.sender import Sender
|
|
from wallabag_kindle_consumer.wallabag import Wallabag
|
|
|
|
StreamHandler(sys.stdout).push_application()
|
|
logger = Logger("kindle-consumer")
|
|
|
|
|
|
def parse_args():
|
|
parser = argparse.ArgumentParser(description="Wallabag-Kindle-Consumer")
|
|
parser.add_argument("--cfg", help="config file", required=False)
|
|
parser.add_argument("--env", help="Read config from env", action="store_true")
|
|
parser.add_argument("--refresher", help="Start token refresher", action="store_true")
|
|
parser.add_argument("--interface", help="Start web interface", action="store_true")
|
|
parser.add_argument("--consumer", help="Start article consumer", action="store_true")
|
|
|
|
return parser.parse_args()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())
|
|
|
|
loop = asyncio.get_event_loop()
|
|
|
|
args = parse_args()
|
|
|
|
config = Config.from_file("config.ini")
|
|
|
|
if 'cfg' in args and args.cfg is not None:
|
|
new = Config.from_file(args.cfg)
|
|
if new is not None:
|
|
config = new
|
|
|
|
if 'env' in args and args.env:
|
|
new = Config.from_env()
|
|
if new is not None:
|
|
config = new
|
|
|
|
wallabag = Wallabag(config)
|
|
sender = Sender(loop, config.smtp_from, config.smtp_host, config.smtp_port, config.smtp_user, config.smtp_passwd)
|
|
|
|
if args.refresher:
|
|
logger.info("Create Refresher")
|
|
refresher = Refresher(config, wallabag, sender)
|
|
loop.create_task(refresher.refresh())
|
|
|
|
if args.consumer:
|
|
logger.info("Create Consumer")
|
|
consumer = Consumer(wallabag, config, sender)
|
|
loop.create_task(consumer.consume())
|
|
|
|
if args.interface:
|
|
logger.info("Create Interface")
|
|
webapp = App(config, wallabag)
|
|
loop.create_task(webapp.register_server(loop))
|
|
|
|
loop.run_forever()
|