scripts/ml-search.py

   1 #! /usr/bin/env python
   2 # coding: utf-8
   3
   4 import argparse
   5 from m_lib.defenc import default_encoding
   6 from m_librarian.search import search_authors, search_books, \
   7     search_extensions, search_genres, search_languages
   8
   9
  10 def _search_authors(args):
  11     values = {}
  12     for column in 'surname', 'name', 'misc':
  13         value = getattr(args, column)
  14         if value:
  15             values[column] = unicode(value, default_encoding)
  16     for author in search_authors(args.search_type, args.case_sensitive,
  17                                  values):
  18         full_name = filter(None,
  19                            (author.surname, author.name, author.misc_name))
  20         full_name = u' '.join(full_name)
  21         print full_name.encode(default_encoding), \
  22             u"(книг: %d)".encode(default_encoding) % author.count
  23
  24
  25 if __name__ == '__main__':
  26     main_parser = argparse.ArgumentParser(description='Search')
  27     main_parser.add_argument('-I', '--case-sensitive',
  28                              action='store_true',
  29                              help='don\'t ignore case '
  30                              '(default is case-insensitive search)')
  31     main_parser.add_argument('-t', '--search-type',
  32                              choices=['exact', 'start', 'substring'],
  33                              help='search type: '
  34                              'exact match, substring at the start '
  35                              '(this is the default), substring anywhere')
  36     subparsers = main_parser.add_subparsers(help='Commands')
  37
  38     parser = subparsers.add_parser('authors', help='Search authors')
  39     parser.add_argument('-s', '--surname', help='search by surname')
  40     parser.add_argument('-n', '--name', help='search by name')
  41     parser.add_argument('-m', '--misc', help='search by misc. name')
  42     parser.set_defaults(func=_search_authors)
  43
  44     args = main_parser.parse_args()
  45     args.func(args)