Initial commit, re-branded from first attempt..
This commit is contained in:
Lance Edgar
commit
7058ebf75e
17 changed files with 871 additions and 0 deletions
249
messkit/commands.py
Normal file
249
messkit/commands.py
Normal file
|
|
@ -0,0 +1,249 @@
|
|||
# -*- coding: utf-8; -*-
|
||||
######################################################################
|
||||
#
|
||||
# messkit -- Generic-ish Data Utility App
|
||||
# Copyright © 2022 Lance Edgar
|
||||
#
|
||||
# This file is part of messkit.
|
||||
#
|
||||
# messkit is free software: you can redistribute it and/or modify it
|
||||
# under the terms of the GNU General Public License as published by
|
||||
# the Free Software Foundation, either version 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# messkit is distributed in the hope that it will be useful, but
|
||||
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
# General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with messkit. If not, see <http://www.gnu.org/licenses/>.
|
||||
#
|
||||
######################################################################
|
||||
"""
|
||||
messkit commands
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
import subprocess
|
||||
|
||||
import sqlalchemy as sa
|
||||
from prompt_toolkit import prompt
|
||||
from prompt_toolkit.styles import Style
|
||||
from rich import print as rprint
|
||||
from alembic.util.messaging import obfuscate_url_pw
|
||||
|
||||
from rattail import commands
|
||||
|
||||
from messkit import __version__
|
||||
|
||||
|
||||
def main(*args):
|
||||
"""
|
||||
Main entry point for messkit command system
|
||||
"""
|
||||
args = list(args or sys.argv[1:])
|
||||
cmd = Command()
|
||||
cmd.run(*args)
|
||||
|
||||
|
||||
class Command(commands.Command):
|
||||
"""
|
||||
Main command for messkit
|
||||
"""
|
||||
name = 'messkit'
|
||||
version = __version__
|
||||
description = "messkit (Generic Data App)"
|
||||
long_description = ''
|
||||
|
||||
|
||||
class Install(commands.Subcommand):
|
||||
"""
|
||||
Install a messkit app
|
||||
"""
|
||||
name = 'install'
|
||||
description = __doc__.strip()
|
||||
|
||||
def run(self, args):
|
||||
|
||||
rprint("\n\t[blue]Welcome to messkit![/blue]")
|
||||
rprint("\n\tThis tool will install and configure a new app.")
|
||||
rprint("\n\t[italic]NB. You should already have created a new database in PostgreSQL.[/italic]")
|
||||
|
||||
# get appdir path
|
||||
appdir = os.path.join(sys.prefix, 'app')
|
||||
appdir = self.basic_prompt('appdir path', appdir)
|
||||
|
||||
# appdir must not yet exist
|
||||
if os.path.exists(appdir):
|
||||
rprint("\n\t[bold red]appdir already exists:[/bold red] {}\n".format(appdir))
|
||||
sys.exit(1)
|
||||
|
||||
# get db info
|
||||
dbhost = self.basic_prompt('postgres host', 'localhost')
|
||||
dbport = self.basic_prompt('postgres port', '5432')
|
||||
dbname = self.basic_prompt('postgres db', 'messkit')
|
||||
dbuser = self.basic_prompt('postgres user', 'rattail')
|
||||
|
||||
# get db password
|
||||
dbpass = None
|
||||
while not dbpass:
|
||||
dbpass = self.basic_prompt('postgres pass', is_password=True)
|
||||
|
||||
# test db connection
|
||||
rprint("\n\ttesting db connection... ", end='')
|
||||
dburl = self.make_db_url(dbhost, dbport, dbname, dbuser, dbpass)
|
||||
error = self.test_db_connection(dburl)
|
||||
if error:
|
||||
rprint("[bold red]cannot connect![/bold red] ..error was:")
|
||||
rprint("\n{}".format(error))
|
||||
rprint("\n\t[bold yellow]aborting mission[/bold yellow]\n")
|
||||
sys.exit(1)
|
||||
rprint("[bold green]good[/bold green]")
|
||||
|
||||
# make the appdir
|
||||
self.app.make_appdir(appdir)
|
||||
|
||||
# make config files
|
||||
context = {
|
||||
'app_title': "messkit",
|
||||
'appdir': appdir,
|
||||
'db_url': dburl,
|
||||
'pyramid_egg': 'messkit',
|
||||
'beaker_key': 'messkit',
|
||||
}
|
||||
rattail_conf = self.app.make_config_file(
|
||||
'rattail-complete', os.path.join(appdir, 'rattail.conf'),
|
||||
**context)
|
||||
quiet_conf = self.app.make_config_file('quiet', appdir)
|
||||
web_conf = self.app.make_config_file(
|
||||
'web-complete', os.path.join(appdir, 'web.conf'),
|
||||
**context)
|
||||
|
||||
rprint("\n\tappdir created at: [bold green]{}[/bold green]".format(appdir))
|
||||
|
||||
bindir = os.path.join(sys.prefix, 'bin')
|
||||
|
||||
schema_installed = False
|
||||
if self.basic_prompt("install db schema?", True, is_bool=True):
|
||||
rprint()
|
||||
|
||||
# install db schema
|
||||
alembic = os.path.join(bindir, 'alembic')
|
||||
cmd = [alembic, '-c', rattail_conf, 'upgrade', 'heads']
|
||||
subprocess.check_call(cmd)
|
||||
schema_installed = True
|
||||
|
||||
rprint("\n\tdb schema installed to: [bold green]{}[/bold green]".format(
|
||||
obfuscate_url_pw(dburl)))
|
||||
|
||||
if self.basic_prompt("create admin user?", True, is_bool=True):
|
||||
|
||||
# get admin credentials
|
||||
username = self.basic_prompt('admin username', 'admin')
|
||||
password = None
|
||||
while not password:
|
||||
password = self.basic_prompt('admin password', is_password=True)
|
||||
if password:
|
||||
confirm = self.basic_prompt('confirm password', is_password=True)
|
||||
if not confirm or confirm != password:
|
||||
rprint("[bold yellow]passwords did not match[/bold yellow]")
|
||||
password = None
|
||||
|
||||
rprint()
|
||||
|
||||
# make admin user
|
||||
rattail = os.path.join(bindir, 'rattail')
|
||||
cmd = [rattail, '-c', quiet_conf, 'make-user', '-A', username,
|
||||
'--password', password]
|
||||
subprocess.check_call(cmd)
|
||||
|
||||
rprint("\n\tadmin user created: [bold green]{}[/bold green]".format(
|
||||
username))
|
||||
|
||||
if self.basic_prompt("make poser dir?", True, is_bool=True):
|
||||
rprint()
|
||||
|
||||
poser_handler = self.app.get_poser_handler()
|
||||
poserdir = poser_handler.make_poser_dir()
|
||||
|
||||
rprint("\n\tposer dir created: [bold green]{}[/bold green]".format(
|
||||
poserdir))
|
||||
|
||||
rprint("\n\t[bold green]initial setup is complete![/bold green]")
|
||||
|
||||
if schema_installed:
|
||||
rprint("\n\tyou can run the web app with this command:")
|
||||
pserve = os.path.join(bindir, 'pserve')
|
||||
rprint("\n\t[blue]{} --reload file+ini:{}[/blue]".format(pserve, web_conf))
|
||||
|
||||
rprint()
|
||||
|
||||
def basic_prompt(self, info, default=None, is_password=False, is_bool=False):
|
||||
|
||||
# message formatting styles
|
||||
style = Style.from_dict({
|
||||
'': '',
|
||||
'bold': 'bold',
|
||||
})
|
||||
|
||||
# build prompt message
|
||||
message = [
|
||||
('', '\n'),
|
||||
('class:bold', info),
|
||||
]
|
||||
if default is not None:
|
||||
if is_bool:
|
||||
message.append(('', ' [{}]: '.format('Y' if default else 'N')))
|
||||
else:
|
||||
message.append(('', ' [{}]: '.format(default)))
|
||||
else:
|
||||
message.append(('', ': '))
|
||||
|
||||
# prompt user for input
|
||||
try:
|
||||
text = prompt(message, style=style, is_password=is_password)
|
||||
except (KeyboardInterrupt, EOFError):
|
||||
rprint("\n\t[bold yellow]operation canceled by user[/bold yellow]\n",
|
||||
file=self.stderr)
|
||||
sys.exit(2)
|
||||
|
||||
if is_bool:
|
||||
if text == '':
|
||||
return default
|
||||
elif text.upper() == 'Y':
|
||||
return True
|
||||
elif text.upper() == 'N':
|
||||
return False
|
||||
rprint("\n\t[bold yellow]ambiguous, please try again[/bold yellow]\n")
|
||||
return self.basic_prompt(info, default, is_bool=True)
|
||||
|
||||
return text or default
|
||||
|
||||
def make_db_url(self, dbhost, dbport, dbname, dbuser, dbpass):
|
||||
try:
|
||||
# newer style
|
||||
from sqlalchemy.engine import URL
|
||||
factory = URL.create
|
||||
except ImportError:
|
||||
# older style
|
||||
from sqlalchemy.engine.url import URL
|
||||
factory = URL
|
||||
|
||||
return factory(drivername='postgresql+psycopg2',
|
||||
username=dbuser,
|
||||
password=dbpass,
|
||||
host=dbhost,
|
||||
port=dbport,
|
||||
database=dbname)
|
||||
|
||||
def test_db_connection(self, url):
|
||||
engine = sa.create_engine(url)
|
||||
|
||||
# check for random table; does not matter if it exists, we
|
||||
# just need to test interaction and this is a neutral way
|
||||
try:
|
||||
engine.has_table('whatever')
|
||||
except Exception as error:
|
||||
return str(error)
|
||||
Loading…
Add table
Add a link
Reference in a new issue