library/library.py

from argparse import ArgumentParser
from library_api import LibraryAPI


if __name__=='__main__':
    parser = ArgumentParser()
    parser.add_argument(
        'command',
        choices=['list', 'add', 'set-title', 'del']
    )
    parser.add_argument('params', nargs='*')
    args = parser.parse_args()

    srv, port, db = 'localhost', 8090, 'odoo_dev_1'
    user, pwd = '779659252@qq.com', '123456'
    api = LibraryAPI(srv, port, db, user, pwd)

    if args.command == 'add':
        for title in args.params:
            new_id = api.create(title)
            print('Book added with ID %d' % new_id)

    if args.command == 'set-title':
        if len(args.params) != 2:
            print('set command requires a title and ID.')
        else:
            bookid, title = int(args.params[0]), args.params[1]
            api.write(title, id)
            print('Title set for Book ID %d' % bookid)

    if args.command == 'list':
        from pprint import pprint

        if len(args.params) == 1:
            pprint(api.search_read(args.params[0]))
        else:
            pprint(api.search_read())

    if args.command == 'del':
        for param in args.params:
            api.unlink(int(param))
            print('Book with ID %s deleted.' % param)