#!/usr/bin/env python # -*- coding: utf-8 -*- import os os.environ['DJANGO_SETTINGS_MODULE']='web.settings' from web.library.models import * from django.db.models import Q from util import * import getopt import sys import search def show_book_or_person(ids, commit_format=False, tmp_file=False): for id in ids: object = get_book_or_person(id) print object def get_book_or_person(id): if id.isdigit(): return get_in_order([Book,Person],id) else: return get_in_order([Person,Book],id) def get_in_order(list, id): for object in list: try: b = object.objects.get(pk=id) return b except object.DoesNotExist: pass return None def search_book_cmd(search_strings, search_description=False): books = search_book(search_strings, search_description) format = '%-13s %-10s %-40s %-30s' for book in books: b_id = book.getid() or '' title = cut_str(book.title, 40) authors = map(lambda p: p.first_name+' '+p.last_name, book.get_authors()) authors_str = cut_str(', '.join(authors), 30) print format % (book.isbn, b_id, title, authors_str) def search_book(search_strings, search_description=False): basic_query = Book.objects.select_related('persons__person','id','alt_titles') for word in search_strings: basic_query=basic_query.filter(Q(title__icontains=word) | Q(subtitle__icontains=word) | Q(alt_titles__alt_title__icontains=word) | Q(id__id__icontains=word) | Q(persons__person__first_name__icontains=word) | Q(persons__person__last_name__icontains=word)) return remove_duplicates(basic_query.all()) def remove_duplicates(list): d = {} for i in list: d[i]=None return d.keys() def search_person_cmd(search_strings, search_description=False): people = search_person(search_strings, search_description) format = '%-20s %-10s %-70s' for person in people: name = cut_str(person.first_name+' '+person.last_name, 20) p_id = cut_str(person.id, 10) books = cut_str(', '.join(map(lambda x: x.book.title, person.books.all())), 70) print format % (name,p_id,books) def search_person(search_strings, search_description=False): basic_query=Person.objects.select_related('books__book__alt_titles') for word in search_strings: basic_query=basic_query.filter(Q(first_name__icontains=word) | Q(last_name__icontains=word) | Q(id__icontains=word) | Q(books__book__isbn__icontains=word) | Q(books__book__title__icontains=word) | Q(books__book__alt_titles__alt_title=word) | Q(books__book__id__id__icontains=word)) return remove_duplicates(basic_query.all()) def commit(filename=None): pass def edit_book_or_person(ids): filename = show_book_or_person(ids, commit_format=True, tmp_file=True) # run_editor(filename) commit(filename) commands = { 'show': { 'args': [('ids', (1,None))], 'options': ['commit_format', 'tmp_file'], 'fun': show_book_or_person }, 'search': { 'args': [('search_strings', (1,None))], 'options': ['search_description'], 'fun': search.search_book_cmd }, 'search-person': { 'args': [('search_strings', (1,None))], 'options': [], 'fun': search_person_cmd }, 'commit': { 'args': [('filename', (0,1))], 'options': [], 'fun': commit }, 'edit': { 'args': [('ids', (1,None))], 'options': [], 'fun': edit_book_or_person } } flags = { 'commit_format': { 'help': 'output data in the format expected by the commit command' }, 'tmp_file': { 'help': 'output data to a new temporary file instead of to stdout' }, 'search_description': { 'help': 'include description field when searching' } } general_options = [] # options applicable to all commands class BadCommandLine(Exception): def __init__(self, msg): Exception.__init__(self, 'Bad command line: ' + msg) def check_command_args(args, command): cmd_decl = commands[command] min_num_args = sum(map(lambda a: a[1][0], cmd_decl['args'])) unlimited = any(map(lambda a: a[1][1] == None, cmd_decl['args'])) if not unlimited: max_num_args = sum(map(lambda a: a[1][1], cmd_decl['args'])) if len(args) < min_num_args: raise BadCommandLine('Too few arguments for command %s (expects at least %d, %d given).' % (command, min_num_args, len(args))) if (not unlimited) and (len(args) < max_num_args): raise BadCommandLine('Too many arguments for command %s (expects at most %d, %d given).' % (command, max_num_args, len(args))) def check_command_opts(opts, command): cmd_decl = commands[command] for opt,val in opts: if ((opt not in cmd_decl['options']) and (opt not in general_options)): raise BadCommandLine('Option %s not applicable to command %s.' % (opt, command)) def assign_command_args(args, command): cmd_decl = commands[command] d = {} i = 0 for param in cmd_decl['args']: pname = param[0] pmin = param[1][0] pmax = param[1][1] if pmax == 1: if i < len(args): d[pname] = args[i] i += 1 else: d[pname] = None else: d[pname] = [] j = 0 while i + j < len(args) and (pmax == None or j < pmax): d[pname].append(args[i + j]) j += 1 i = i + j return d def assign_command_opts(opts, command): d = {} for option, value in opts: if option in flags: d[option] = True else: d[option] = value return d def parse_cmdline(args): def getopt_option_name(internal_name): return internal_name.replace('_', '-') def internal_option_name(getopt_ret_name): return getopt_ret_name[2:].replace('-', '_') option_names = map(getopt_option_name, flags) options, args = getopt.getopt(args, '', option_names) if len(args) == 0: raise BadCommandLine('No command specified.') cmd_name = args[0] if cmd_name not in commands: raise BadCommandLine('Nonexisting command %s.' % cmd_name) cmd_decl = commands[cmd_name] cmd_args = args[1:] cmd_opts = map(lambda (opt,val): (internal_option_name(opt), val), options) check_command_args(cmd_args, cmd_name) check_command_opts(cmd_opts, cmd_name) return { 'command': cmd_name, 'args': combine_dicts(assign_command_args(cmd_args, cmd_name), assign_command_opts(cmd_opts, cmd_name)) } def invoke_command(command, args): cmd_decl = commands[command] cmd_decl['fun'](**args) def combine_dicts(*dicts): res = {} for d in dicts: res.update(d) return res def open_file(filename): if os.path.exists(filename): os.system("%s %s || /usr/bin/env vi %s" %(os.getenv("EDITOR"), filename, filename)) else: exit("Error: %s: File does not exist!" % filename) return open(filename, 'r').readlines() cmdline_parsed = parse_cmdline(sys.argv[1:]) print 'command line parsed to:', cmdline_parsed invoke_command(cmdline_parsed['command'], cmdline_parsed['args'])