yage
76a746a53e
=====================================================================
Notice the major version change which comes with breaking changes to
2.x! Reconstructs "library" functions for both python and zsh scwrypts,
with changes to virtualenv naming conventions (you'll need to refresh
all virtualenv with the appropriate scwrypt).
--- Changes ------------------------------
- changed a naming convention across zsh scripts, particularly
removing underscores where there is no need to avoid naming clash
(e.g. 'zsh/lib/utils/io.zsh' renames '__STATUS' to 'STATUS')
- moved clients reliant on py.lib.http to the py.lib.http module
- python scripts now rely on py.lib.scwrypts.execute
- updated package.json in zx scripts to include `type = module`
- 'scwrypts --list' commandline argument now includes additional
relevant data for each scwrypt
- environment variables no longer add themselves to be staged in the
'.env.template'
--- New Features -------------------------
- new 'use' syntax for disjoint import within zsh scripts; took me
a very long time to convince myself this would be necessary
- introduced scwrypt "groups" to allow portable module creation;
(i.e. ability add your own scripts from another repo!)
- py.lib.scwrypts.io provides a combined IO stream for quick, hybrid
use of input/output files and stdin/stdout
- py.lib.fzf provides a wrapper to provide similar functionality to
zsh/utils/io.zsh including fzf_(head|tail)
- improved efficiency of various scwrypts; notably reducing runtime
of scwrypts/environment sync
- improved scwrypts CLI by adding new options for exact scwrypt
matching, better filtering, and prettier/more-detailed interfaces
--- New Scripts --------------------------
- py/twilio )
basic SMS integration with twilio
- send-sms
- py/directus )
interactive directus GET query
- get-items
- py/discord )
post message to discord channel or webhook
- post-message
87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
from contextlib import contextmanager
|
|
from pathlib import Path
|
|
from sys import stdin, stdout, stderr
|
|
|
|
from py.lib.scwrypts.getenv import getenv
|
|
|
|
|
|
@contextmanager
|
|
def get_stream(filename=None, mode='r', encoding='utf-8', verbose=False, **kwargs):
|
|
allowed_modes = {'r', 'w', 'w+'}
|
|
|
|
if mode not in allowed_modes:
|
|
raise ValueError(f'mode "{mode}" not supported modes (must be one of {allowed_modes})')
|
|
|
|
is_read = mode == 'r'
|
|
|
|
if filename is not None:
|
|
|
|
if verbose:
|
|
print(f'opening file {filename} for {"read" if is_read else "write"}', file=stderr)
|
|
|
|
if filename[0] not in {'/', '~'}:
|
|
filename = Path(f'{getenv("EXECUTION_DIR")}/{filename}').resolve()
|
|
with open(filename, mode=mode, encoding=encoding, **kwargs) as stream:
|
|
yield stream
|
|
|
|
else:
|
|
if verbose:
|
|
print('using stdin for read' if is_read else 'using stdout for write', file=stderr)
|
|
|
|
yield stdin if is_read else stdout
|
|
|
|
if not is_read:
|
|
stdout.flush()
|
|
|
|
|
|
def add_io_arguments(parser, toggle_input=True, toggle_output=True):
|
|
if toggle_input:
|
|
parser.add_argument(
|
|
'-i', '--input-file',
|
|
dest = 'input_file',
|
|
default = None,
|
|
help = 'path to input file; omit for stdin',
|
|
required = False,
|
|
)
|
|
|
|
if toggle_output:
|
|
parser.add_argument(
|
|
'-o', '--output-file',
|
|
dest = 'output_file',
|
|
default = None,
|
|
help = 'path to output file; omit for stdout',
|
|
required = False,
|
|
)
|
|
|
|
|
|
@contextmanager
|
|
def get_combined_stream(input_file=None, output_file=None):
|
|
with get_stream(input_file, 'r') as input_stream, get_stream(output_file, 'w+') as output_stream:
|
|
yield CombinedStream(input_stream, output_stream)
|
|
|
|
|
|
class CombinedStream:
|
|
def __init__(self, input_stream, output_stream):
|
|
self.input = input_stream
|
|
self.output = output_stream
|
|
|
|
def read(self, *args, **kwargs):
|
|
return self.input.read(*args, **kwargs)
|
|
|
|
def readline(self, *args, **kwargs):
|
|
return self.input.readline(*args, **kwargs)
|
|
|
|
def readlines(self, *args, **kwargs):
|
|
return self.input.readlines(*args, **kwargs)
|
|
|
|
def write(self, *args, **kwargs):
|
|
return self.output.write(*args, **kwargs)
|
|
|
|
def writeline(self, line):
|
|
x = self.output.write(f'{line}\n')
|
|
self.output.flush()
|
|
return x
|
|
|
|
def writelines(self, *args, **kwargs):
|
|
return self.output.writelines(*args, **kwargs)
|