worblehat-old/python/worblehat.py

#!/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)
            # TODO: skriv ut mer informasjon
	    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): # TODO
    pass

def edit_book_or_person(ids):
    filename = show_book_or_person(ids, commit_format=True, tmp_file=True)
#    run_editor(filename) # TODO: må implementeres
    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'])