summaryrefslogtreecommitdiffstats
path: root/src/python/abcpy_test.py
blob: 912cae9f5c6ae128eab81b075b7d4af38befbc13 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
# You can use 'from pyabc import *' and then not need the pyabc. prefix everywhere
import pyabc

# A new command is just a function that accepts a list of string arguments
#   The first argument is always the name of the command
#   It MUST return an integer. -1: user quits, -2: error. Return 0 for success.

# a simple command that just prints its arguments and returns success
def pytest1_cmd(args):
    print args
    return 0

# registers the command:
#   The first argument is the function
#   The second argument is the category (mainly for the ABC help command)
#   The third argument is the new command name
#   Keet the fourth argument 0, or consult with Alan
pyabc.add_abc_command(pytest1_cmd, "Python-Test", "pytest1", 0)

# a simple command that just prints its arguments and runs the command 'scorr -h'
def pytest2_cmd(args):
    print args
    pyabc.run_command('scorr -h')
    return 0

pyabc.add_abc_command(pytest2_cmd, "Python-Test", "pytest2", 0)

# Now a more complicated command with argument parsing
# This command gets two command line arguments -c and -v. -c cmd runs the command 'cmd -h' and -v prints the python version
# for more details see the optparse module: http://docs.python.org/library/optparse.html

import optparse

def pytest3_cmd(args):
    usage = "usage: %prog [options]"
    
    parser = optparse.OptionParser(usage)
    
    parser.add_option("-c", "--cmd", dest="cmd", help="command to ask help for")
    parser.add_option("-v", "--version", action="store_true", dest="version", help="display Python Version")

    options, args = parser.parse_args(args)
    
    if options.version:
        print sys.version
        return 0
    
    if options.cmd:
        pyabc.run_command("%s -h"%options.cmd)
        return 0
    
    return 0

pyabc.add_abc_command(pytest3_cmd, "Python-Test", "pytest3", 0)