mirror of
https://github.com/aljazceru/nutshell.git
synced 2025-12-20 02:24:20 +01:00
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
import click
|
|
import uvicorn
|
|
|
|
from core.settings import (
|
|
MINT_SERVER_HOST,
|
|
MINT_SERVER_PORT,
|
|
)
|
|
|
|
|
|
@click.command(
|
|
context_settings=dict(
|
|
ignore_unknown_options=True,
|
|
allow_extra_args=True,
|
|
)
|
|
)
|
|
@click.option("--port", default=MINT_SERVER_PORT, help="Port to listen on")
|
|
@click.option("--host", default=MINT_SERVER_HOST, help="Host to run mint on")
|
|
@click.option("--ssl-keyfile", default=None, help="Path to SSL keyfile")
|
|
@click.option("--ssl-certfile", default=None, help="Path to SSL certificate")
|
|
@click.pass_context
|
|
def main(
|
|
ctx,
|
|
port: int = MINT_SERVER_PORT,
|
|
host: str = MINT_SERVER_HOST,
|
|
ssl_keyfile: str = None,
|
|
ssl_certfile: str = None,
|
|
):
|
|
"""Launched with `poetry run mint` at root level"""
|
|
# this beautiful beast parses all command line arguments and passes them to the uvicorn server
|
|
d = dict()
|
|
for a in ctx.args:
|
|
item = a.split("=")
|
|
if len(item) > 1: # argument like --key=value
|
|
print(a, item)
|
|
d[item[0].strip("--").replace("-", "_")] = (
|
|
int(item[1]) # need to convert to int if it's a number
|
|
if item[1].isdigit()
|
|
else item[1]
|
|
)
|
|
else:
|
|
d[a.strip("--")] = True # argument like --key
|
|
|
|
config = uvicorn.Config(
|
|
"mint.app:app",
|
|
port=port,
|
|
host=host,
|
|
ssl_keyfile=ssl_keyfile,
|
|
ssl_certfile=ssl_certfile,
|
|
**d,
|
|
)
|
|
server = uvicorn.Server(config)
|
|
server.run()
|