Compare commits
2 Commits
58e4e42b30
...
e36e5a1428
Author | SHA1 | Date |
---|---|---|
Ales (Shagi) Zabala Alava | e36e5a1428 | |
Ales (Shagi) Zabala Alava | 261c8b6275 |
|
@ -1,11 +1,18 @@
|
||||||
import argparse
|
import argparse
|
||||||
|
import configparser
|
||||||
|
import logging
|
||||||
|
import traceback
|
||||||
import sys
|
import sys
|
||||||
|
from pathlib import Path
|
||||||
try:
|
try:
|
||||||
from importlib import metadata
|
from importlib import metadata
|
||||||
except ImportError: # Python <3.8
|
except ImportError: # Python <3.8
|
||||||
import importlib_metadata as metadata
|
import importlib_metadata as metadata
|
||||||
|
|
||||||
|
|
||||||
|
CONFIG_FILE = '~/.negromate/config.ini'
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
"""
|
"""
|
||||||
Build parser for all the commands and launch appropiate command.
|
Build parser for all the commands and launch appropiate command.
|
||||||
|
@ -15,25 +22,31 @@ def main():
|
||||||
* name: String with the command name. Will be used for
|
* name: String with the command name. Will be used for
|
||||||
argparse subcommand.
|
argparse subcommand.
|
||||||
* help_text: String with the help text.
|
* help_text: String with the help text.
|
||||||
|
* initial_config: Dict for initial configuration of commands.
|
||||||
* options: Function to build the parser of the command. Takes
|
* options: Function to build the parser of the command. Takes
|
||||||
one parametter, the argparser parser instance for this
|
two parametters, the argparser parser instance for this
|
||||||
subcommand.
|
subcommand and the ConfigParser instance with all the
|
||||||
* run: Function that runs the actual command. Takes one
|
configuration.
|
||||||
parametter, the Namespace with the arguments.
|
* run: Function that runs the actual command. Takes two
|
||||||
|
parametters, the argparse Namespace with the arguments and
|
||||||
|
the ConfigParser with all the configuration.
|
||||||
|
|
||||||
Minimal module example:
|
Minimal module example:
|
||||||
|
|
||||||
# hello_world.py
|
# hello_world.py
|
||||||
name = 'hello'
|
name = 'hello'
|
||||||
help_text = 'Sample command'
|
help_text = 'Sample command'
|
||||||
|
initial_config = {
|
||||||
|
'who': 'World',
|
||||||
|
}
|
||||||
|
|
||||||
def options(parser):
|
def options(parser, config, **kwargs):
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
'-w', '--who', default='World'
|
'-w', '--who', default=config['hello']['who'],
|
||||||
help="Who to say hello, defaults to 'World'"
|
help="Who to say hello, defaults to '{}'".format(config['hello']['who'])
|
||||||
)
|
)
|
||||||
|
|
||||||
def run(args):
|
def run(args, **kwargs):
|
||||||
print("Hello {}".format(args.who))
|
print("Hello {}".format(args.who))
|
||||||
|
|
||||||
To add more commands to negromate register 'negromate.commands'
|
To add more commands to negromate register 'negromate.commands'
|
||||||
|
@ -50,22 +63,52 @@ def main():
|
||||||
args = sys.argv.copy()
|
args = sys.argv.copy()
|
||||||
sys.argv = args[:1]
|
sys.argv = args[:1]
|
||||||
|
|
||||||
|
# Load commands from entry_point
|
||||||
entry_points = metadata.entry_points().get('negromate.commands', [])
|
entry_points = metadata.entry_points().get('negromate.commands', [])
|
||||||
for entry_point in entry_points:
|
for entry_point in entry_points:
|
||||||
commands.append(entry_point.load())
|
try:
|
||||||
|
command = entry_point.load()
|
||||||
|
except Exception as e:
|
||||||
|
traceback.print_exc()
|
||||||
|
print(e)
|
||||||
|
continue
|
||||||
|
commands.append(command)
|
||||||
|
|
||||||
|
# Load initial configuration for commands
|
||||||
|
initial_config = {
|
||||||
|
'global': {
|
||||||
|
'song_folder': '~/negro_mate/bideoak/',
|
||||||
|
'lyrics_file': '~/negro_mate/libreto/libreto.pdf',
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for command in commands:
|
||||||
|
if hasattr(command, 'initial_config'):
|
||||||
|
initial_config[command.name] = command.initial_config
|
||||||
|
|
||||||
|
# Load configuration
|
||||||
|
config = configparser.ConfigParser()
|
||||||
|
config.read_dict(initial_config)
|
||||||
|
config.read(Path(CONFIG_FILE).expanduser())
|
||||||
|
|
||||||
|
# Build parser
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
|
parser.add_argument(
|
||||||
|
'-v', '--verbose', action='store_true',
|
||||||
|
help="Display informational messages.")
|
||||||
parser.set_defaults(command=None)
|
parser.set_defaults(command=None)
|
||||||
subparsers = parser.add_subparsers()
|
subparsers = parser.add_subparsers()
|
||||||
for command in commands:
|
for command in commands:
|
||||||
command_parser = subparsers.add_parser(command.name, help=command.help_text)
|
command_parser = subparsers.add_parser(command.name, help=command.help_text)
|
||||||
command_parser.set_defaults(command=command.name)
|
command_parser.set_defaults(command=command.name)
|
||||||
command.options(command_parser)
|
command.options(parser=command_parser, config=config)
|
||||||
|
|
||||||
|
# Run command
|
||||||
args = parser.parse_args(args[1:])
|
args = parser.parse_args(args[1:])
|
||||||
|
if args.verbose:
|
||||||
|
logging.basicConfig(level=logging.DEBUG)
|
||||||
if args.command is None:
|
if args.command is None:
|
||||||
parser.print_usage()
|
parser.print_usage()
|
||||||
else:
|
else:
|
||||||
for command in commands:
|
for command in commands:
|
||||||
if args.command == command.name:
|
if args.command == command.name:
|
||||||
command.run(args)
|
command.run(args=args, config=config)
|
||||||
|
|
|
@ -0,0 +1,21 @@
|
||||||
|
import sys
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
name = 'config'
|
||||||
|
help_text = 'Write the configuration'
|
||||||
|
initial_config = {
|
||||||
|
'file': '~/.negromate/config.ini',
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
def options(parser, config, **kwargs):
|
||||||
|
parser.add_argument(
|
||||||
|
'-f', '--file', type=Path,
|
||||||
|
default=config['config']['file'],
|
||||||
|
help="Configuration file, defaults to {}".format(
|
||||||
|
config['config']['file']))
|
||||||
|
|
||||||
|
|
||||||
|
def run(args, config, **kwargs):
|
||||||
|
with args.file.expanduser().open('w') as f:
|
||||||
|
config.write(f)
|
|
@ -7,14 +7,16 @@ name = 'songs'
|
||||||
help_text = 'Update song database'
|
help_text = 'Update song database'
|
||||||
|
|
||||||
|
|
||||||
def options(parser):
|
def options(parser, config, **kwargs):
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
'song_folder', type=Path,
|
'-s', '--song_folder', type=Path,
|
||||||
help="Folder with the song database.")
|
default=config['global']['song_folder'],
|
||||||
|
help="Folder with the song database, defaults to {}".format(
|
||||||
|
config['global']['song_folder']))
|
||||||
|
|
||||||
|
|
||||||
def run(args):
|
def run(args, **kwargs):
|
||||||
songs, pending_songs = load_songs(args.song_folder)
|
songs, pending_songs = load_songs(args.song_folder.expanduser())
|
||||||
print(
|
print(
|
||||||
"#######\n"
|
"#######\n"
|
||||||
" Songs\n"
|
" Songs\n"
|
||||||
|
|
|
@ -6,7 +6,7 @@ name = 'thumbnail'
|
||||||
help_text = 'Generate cover and thumbnail for a video.'
|
help_text = 'Generate cover and thumbnail for a video.'
|
||||||
|
|
||||||
|
|
||||||
def options(parser):
|
def options(parser, config, **kwargs):
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
'video', help="Video of the song.", type=Path)
|
'video', help="Video of the song.", type=Path)
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
|
@ -14,7 +14,7 @@ def options(parser):
|
||||||
help='Take snapshot at this second.')
|
help='Take snapshot at this second.')
|
||||||
|
|
||||||
|
|
||||||
def run(args):
|
def run(args, **kwargs):
|
||||||
video = args.video
|
video = args.video
|
||||||
cover = video.parent / 'cover.jpg'
|
cover = video.parent / 'cover.jpg'
|
||||||
thumbnail = video.parent / 'thumb.jpg'
|
thumbnail = video.parent / 'thumb.jpg'
|
||||||
|
|
|
@ -32,6 +32,7 @@ console_scripts =
|
||||||
negromate = negromate.songs.commands:main
|
negromate = negromate.songs.commands:main
|
||||||
negromate.commands =
|
negromate.commands =
|
||||||
songs = negromate.songs.commands.songs
|
songs = negromate.songs.commands.songs
|
||||||
|
config = negromate.songs.commands.config
|
||||||
thumbnail = negromate.songs.commands.thumbnail
|
thumbnail = negromate.songs.commands.thumbnail
|
||||||
|
|
||||||
[bdist_wheel]
|
[bdist_wheel]
|
||||||
|
|
Loading…
Reference in New Issue