Basic infrastructure for writing scripts
Project description
Python is an excellent language that makes writing scripts very straightforward. Over the course of writing many scripts, we realized that we were doing some things over and over like creating a logger and accepting command line arguments. Base script is a very simple abstraction that takes care of setting up logging and other basics so you can focus on your application specific logic.
Here are some facilities that Base Script offers:
Logging
Accepting command-line arguments using argparse
Installation
pip install basescript
Usage
Here is a simple example to get started
Hello World
helloworld.py
from basescript import BaseScript
class HelloWorld(BaseScript):
def run(self):
print "Hello world"
if __name__ == '__main__':
HelloWorld().run()
NOTE: all examples showcased here are available under the
``examples`` directory
Run the above by doing:
python helloworld.py
Run script with log level set to DEBUG
python helloworld.py --log-level DEBUG
Run script with custom log file
python helloworld.py --log-level DEBUG --log mylog
Command line args, Using the logger
The following is a more involved example
adder.py
from basescript import BaseScript
class Adder(BaseScript):
# The following specifies the script description so that it be used
# as a part of the usage doc when --help option is used during running.
DESC = 'Adds numbers'
def init(self):
'''
We can put whatever script initialization we need for our script
over here. This is preferred to overriding __init__
'''
self.a = 10
self.b = 20
def define_args(self, parser):
parser.add_argument('c', type=int, help='Number to add')
def run(self):
self.log.info("Starting run of script ...")
print self.a + self.b + self.args.c
self.log.info("Script is done")
if __name__ == '__main__':
Adder().run()
python adder.py --help
usage: adder.py [-h] [--name NAME] [--log LOG]
[--log-level LOG_LEVEL] [--quiet]
c
Adds numbers
positional arguments:
c Number to add
optional arguments:
-h, --help show this help message and exit
--name NAME Name to identify this instance
--log LOG Name of log file
--log-level LOG_LEVEL
Logging level as picked from the logging module
--quiet
Run the script now to see the intended output
python adder.py 30
60
Run the same with info and higher level logs enabled
python adder.py --log-level INFO 30
2016-04-10 13:48:27,356 INFO Starting run of script ...
60
2016-04-10 13:48:27,356 INFO Script is done