| 1 | #!/usr/bin/env python | 1 | #!/usr/bin/env python |
|---|
| 2 | | 2 | |
|---|
| 3 | import cmd | 3 | import cmd |
|---|
| 4 | import os | 4 | import os |
|---|
| 5 | import StringIO | 5 | import StringIO |
|---|
| 6 | | 6 | |
|---|
| 7 | class ShakespeareAdmin(cmd.Cmd): | 7 | class ShakespeareAdmin(cmd.Cmd): |
|---|
| 8 | """ | 8 | """ |
|---|
| 9 | TODO: self.verbose option and associated self._print | 9 | TODO: self.verbose option and associated self._print |
|---|
| 10 | """ | 10 | """ |
|---|
| 11 | | 11 | |
|---|
| 12 | prompt = 'The Bard > ' | 12 | prompt = 'The Bard > ' |
|---|
| 13 | | 13 | |
|---|
| 14 | def run_interactive(self, line=None): | 14 | def run_interactive(self, line=None): |
|---|
| 15 | """Run an interactive session. | 15 | """Run an interactive session. |
|---|
| 16 | """ | 16 | """ |
|---|
| 17 | print 'Welcome to shakespeare-admin interactive mode\n' | 17 | print 'Welcome to shakespeare-admin interactive mode\n' |
|---|
| 18 | self.do_about() | 18 | self.do_about() |
|---|
| 19 | print 'Type: "?" or "help" for help on commands.\n' | 19 | print 'Type: "?" or "help" for help on commands.\n' |
|---|
| 20 | while 1: | 20 | while 1: |
|---|
| 21 | try: | 21 | try: |
|---|
| 22 | self.cmdloop() | 22 | self.cmdloop() |
|---|
| 23 | break | 23 | break |
|---|
| 24 | except KeyboardInterrupt: | 24 | except KeyboardInterrupt: |
|---|
| 25 | raise | 25 | raise |
|---|
| 26 | | 26 | |
|---|
| 27 | def do_help(self, line=None): | 27 | def do_help(self, line=None): |
|---|
| 28 | cmd.Cmd.do_help(self, line) | 28 | cmd.Cmd.do_help(self, line) |
|---|
| 29 | | 29 | |
|---|
| 30 | def do_about(self, line=None): | 30 | def do_about(self, line=None): |
|---|
| 31 | import shakespeare | 31 | import shakespeare |
|---|
| 32 | version = shakespeare.__version__ | 32 | version = shakespeare.__version__ |
|---|
| 33 | about = \ | 33 | about = \ |
|---|
| 34 | '''Open Shakespeare version %s. Copyright the Open Knowledge Foundation. | 34 | '''Open Shakespeare version %s. Copyright the Open Knowledge Foundation. |
|---|
| 35 | Open Shakespeare is open-knowledge and open-source. See COPYING for details. | 35 | Open Shakespeare is open-knowledge and open-source. See COPYING for details. |
|---|
| 36 | ''' % version | 36 | ''' % version |
|---|
| 37 | print about | 37 | print about |
|---|
| 38 | | 38 | |
|---|
| 39 | def do_quit(self, line=None): | 39 | def do_quit(self, line=None): |
|---|
| 40 | sys.exit() | 40 | sys.exit() |
|---|
| 41 | | 41 | |
|---|
| 42 | def do_EOF(self, *args): | 42 | def do_EOF(self, *args): |
|---|
| 43 | print '' | 43 | print '' |
|---|
| 44 | sys.exit() | 44 | sys.exit() |
|---|
| 45 | | 45 | |
|---|
| 46 | # ================= | 46 | # ================= |
|---|
| 47 | # Commands | 47 | # Commands |
|---|
| 48 | | 48 | |
|---|
| 49 | def do_db(self, line=None): | 49 | def do_db(self, line=None): |
|---|
| 50 | actions = [ 'create', 'clean', 'rebuild' ] | 50 | actions = [ 'create', 'clean', 'rebuild' ] |
|---|
| 51 | if line is None or line not in actions: | 51 | if line is None or line not in actions: |
|---|
| 52 | self.help_db() | 52 | self.help_db() |
|---|
| 53 | return 1 | 53 | return 1 |
|---|
| 54 | import shakespeare.dm | 54 | import shakespeare.dm |
|---|
| 55 | shakespeare.dm.__dict__[line+'db']() | 55 | shakespeare.dm.__dict__[line+'db']() |
|---|
| 56 | | 56 | |
|---|
| 57 | def help_db(self, line=None): | 57 | def help_db(self, line=None): |
|---|
| 58 | usage = \ | 58 | usage = \ |
|---|
| 59 | '''db <action> | 59 | '''db <action> |
|---|
| 60 | | 60 | |
|---|
| 61 | Where action is one of create, clean, rebuild.''' | 61 | Where action is one of create, clean, rebuild.''' |
|---|
| 62 | print usage | 62 | print usage |
|---|
| 63 | | 63 | |
|---|
| 64 | def do_gutenberg(self, line=None): | 64 | def do_gutenberg(self, line=None): |
|---|
| 65 | import shakespeare.gutenberg | 65 | import shakespeare.gutenberg |
|---|
| 66 | helper = shakespeare.gutenberg.Helper(verbose=True) | 66 | helper = shakespeare.gutenberg.Helper(verbose=True) |
|---|
| 67 | if not line: | 67 | if not line: |
|---|
| 68 | helper.execute() | 68 | helper.execute() |
|---|
| 69 | elif line == 'print_index': | 69 | elif line == 'print_index': |
|---|
| 70 | import pprint | 70 | import pprint |
|---|
| 71 | pprint.pprint(helper.get_index()) | 71 | pprint.pprint(helper.get_index()) |
|---|
| 72 | else: | 72 | else: |
|---|
| 73 | msg = 'Unknown argument %s' % line | 73 | msg = 'Unknown argument %s' % line |
|---|
| 74 | raise Exception(msg) | 74 | raise Exception(msg) |
|---|
| 75 | | 75 | |
|---|
| 76 | def help_gutenberg(self, line=None): | 76 | def help_gutenberg(self, line=None): |
|---|
| 77 | usage = \ | 77 | usage = \ |
|---|
| 78 | """ | 78 | """ |
|---|
| 79 | Download and process all Project Gutenberg shakespeare texts""" | 79 | Download and process all Project Gutenberg shakespeare texts""" |
|---|
| 80 | print usage | 80 | print usage |
|---|
| 81 | | 81 | |
|---|
| 82 | def do_moby(self, line=None): | 82 | def do_moby(self, line=None): |
|---|
| 83 | import shakespeare.moby | 83 | import shakespeare.moby |
|---|
| 84 | helper = shakespeare.moby.Helper(verbose=True) | 84 | helper = shakespeare.moby.Helper(verbose=True) |
|---|
| 85 | if not line: | 85 | if not line: |
|---|
| 86 | helper.execute() | 86 | helper.execute() |
|---|
| 87 | elif line == 'print_index': | 87 | elif line == 'print_index': |
|---|
| 88 | import pprint | 88 | import pprint |
|---|
| 89 | pprint.pprint(helper.get_index()) | 89 | pprint.pprint(helper.get_index()) |
|---|
| 90 | else: | 90 | else: |
|---|
| 91 | msg = 'Unknown argument %s' % line | 91 | msg = 'Unknown argument %s' % line |
|---|
| 92 | raise Exception(msg) | 92 | raise Exception(msg) |
|---|
| 93 | | 93 | |
|---|
| 94 | def help_moby(self, line=None): | 94 | def help_moby(self, line=None): |
|---|
| 95 | usage = \ | 95 | usage = \ |
|---|
| 96 | ''' | 96 | ''' |
|---|
| 97 | Download and process all Moby/Bosak shakespeare texts''' | 97 | Download and process all Moby/Bosak shakespeare texts''' |
|---|
| 98 | print usage | 98 | print usage |
|---|
| 99 | | 99 | |
|---|
| 100 | def _init_index(self): | 100 | def _init_index(self): |
|---|
| 101 | import shakespeare.index | 101 | import shakespeare.index |
|---|
| 102 | self._index = shakespeare.index.all | 102 | self._index = shakespeare.index.all |
|---|
| 103 | | 103 | |
|---|
| 104 | def _filter_index(self, line): | 104 | def _filter_index(self, line): |
|---|
| 105 | """Filter items in index return only those whose id (url) is in line | 105 | """Filter items in index return only those whose id (url) is in line |
|---|
| 106 | If line is empty or None return all items | 106 | If line is empty or None return all items |
|---|
| 107 | """ | 107 | """ |
|---|
| 108 | if line: | 108 | if line: |
|---|
| 109 | textsToAdd = [] | 109 | textsToAdd = [] |
|---|
| 110 | textNames = line.split() | 110 | textNames = line.split() |
|---|
| 111 | for item in self._index: | 111 | for item in self._index: |
|---|
| 112 | if item.name in textNames: | 112 | if item.name in textNames: |
|---|
| 113 | textsToAdd.append(item) | 113 | textsToAdd.append(item) |
|---|
| 114 | return textsToAdd | 114 | return textsToAdd |
|---|
| 115 | else: | 115 | else: |
|---|
| 116 | self._init_index() | 116 | self._init_index() |
|---|
| 117 | return self._index | 117 | return self._index |
|---|
| 118 | | 118 | |
|---|
| 119 | def do_print_index(self, line): | 119 | def do_print_index(self, line): |
|---|
| 120 | self._init_index() | 120 | self._init_index() |
|---|
| 121 | header = \ | 121 | header = \ |
|---|
| 122 | ''' +-------------------+ | 122 | ''' +-------------------+ |
|---|
| 123 | | Index of Material | | 123 | | Index of Material | |
|---|
| 124 | +-------------------+ | 124 | +-------------------+ |
|---|
| 125 | | 125 | |
|---|
| 126 | ''' | 126 | ''' |
|---|
| 127 | print header | 127 | print header |
|---|
| 128 | for row in self._index: | 128 | for row in self._index: |
|---|
| 129 | print row.name.ljust(35), row.title | 129 | print row.name.ljust(35), row.title |
|---|
| 130 | | 130 | |
|---|
| 131 | def help_print_index(self, line=None): | 131 | def help_print_index(self, line=None): |
|---|
| 132 | usage = \ | 132 | usage = \ |
|---|
| 133 | '''Print index of Shakespeare texts to stdout''' | 133 | '''Print index of Shakespeare texts to stdout''' |
|---|
| 134 | print usage | 134 | print usage |
|---|
| 135 | | 135 | |
|---|
| 136 | def do_make_concordance(self, line=None): | 136 | def do_make_concordance(self, line=None): |
|---|
| 137 | self._init_index() | 137 | self._init_index() |
|---|
| 138 | print 'Making concordance (this may take some time ...):' | 138 | print 'Making concordance (this may take some time ...):' |
|---|
| 139 | from shakespeare.concordance import ConcordanceBuilder | 139 | from shakespeare.concordance import ConcordanceBuilder |
|---|
| 140 | import time | 140 | import time |
|---|
| 141 | start = end = 0 | 141 | start = end = 0 |
|---|
| 142 | start = time.time() | 142 | start = time.time() |
|---|
| 143 | cc = ConcordanceBuilder() | 143 | cc = ConcordanceBuilder() |
|---|
| 144 | textsToAdd = [] | 144 | textsToAdd = [] |
|---|
| 145 | if line is not None: | 145 | if line is not None: |
|---|
| 146 | textsToAdd = self._filter_index(line) | 146 | textsToAdd = self._filter_index(line) |
|---|
| 147 | else: | 147 | else: |
|---|
| 148 | def gut_non_folio(material): | 148 | def gut_non_folio(material): |
|---|
| 149 | return '_gut' in material.name and 'gut_f' not in material.name | 149 | return '_gut' in material.name and 'gut_f' not in material.name |
|---|
| 150 | textsToAdd = filter(gut_non_folio, self._index) | 150 | textsToAdd = filter(gut_non_folio, self._index) |
|---|
| 151 | for item in textsToAdd: | 151 | for item in textsToAdd: |
|---|
| 152 | print 'Adding: %s (%s)' % (item.name, item.title) | 152 | print 'Adding: %s (%s)' % (item.name, item.title) |
|---|
| 153 | cc.add_text(item.name) | 153 | cc.add_text(item.name) |
|---|
| 154 | end = time.time() | 154 | end = time.time() |
|---|
| 155 | timetaken = end - start | 155 | timetaken = end - start |
|---|
| 156 | print 'Finished. Time taken was %ss' % timetaken | 156 | print 'Finished. Time taken was %ss' % timetaken |
|---|
| 157 | | 157 | |
|---|
| 158 | def help_make_concordance(self, line=None): | 158 | def help_make_concordance(self, line=None): |
|---|
| 159 | usage = \ | 159 | usage = \ |
|---|
| 160 | '''Create a concordance | 160 | '''Create a concordance |
|---|
| 161 | | 161 | |
|---|
| 162 | If no arguments supplied then use all non-folio gutenberg shakespeare texts. | 162 | If no arguments supplied then use all non-folio gutenberg shakespeare texts. |
|---|
| 163 | Otherwise arguments should be a space seperated list of work name ids | 163 | Otherwise arguments should be a space seperated list of work name ids |
|---|
| 164 | ''' | 164 | ''' |
|---|
| 165 | print usage | 165 | print usage |
|---|
| 166 | | 166 | |
|---|
| 167 | def do_init(self, line=None): | 167 | def do_init(self, line=None): |
|---|
| 168 | self.do_gutenberg(line) | 168 | self.do_gutenberg(line) |
|---|
| 169 | self.do_moby(line) | 169 | self.do_moby(line) |
|---|
| 170 | self.do_make_concordance(line) | 170 | self.do_make_concordance(line) |
|---|
| 171 | | 171 | |
|---|
| 172 | def help_init(self, line=None): | 172 | def help_init(self, line=None): |
|---|
| 173 | usage = \ | 173 | usage = \ |
|---|
| 174 | '''Convenience function that sets everything up by running: | 174 | '''Convenience function that sets everything up by running: |
|---|
| 175 | 1. gutenberg | 175 | 1. gutenberg |
|---|
| 176 | 2. moby | 176 | 2. moby |
|---|
| 177 | 3. make_concordance''' | 177 | 3. make_concordance''' |
|---|
| 178 | print usage | 178 | print usage |
|---|
| 179 | | 179 | |
|---|