Google

Edit File: urlgrabber

#!/usr/bin/python -t # This library is free software; you can redistribute it and/or # modify it under the terms of the GNU Lesser General Public # License as published by the Free Software Foundation; either # version 2.1 of the License, or (at your option) any later version. # # This library is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU # Lesser General Public License for more details. # # You should have received a copy of the GNU Lesser General Public # License along with this library; if not, write to the # Free Software Foundation, Inc., # 59 Temple Place, Suite 330, # Boston, MA 02111-1307 USA # This file is part of urlgrabber, a high-level cross-protocol url-grabber # Copyright 2002-2006 Michael D. Stenner, Ryan Tomayko """NAME urlgrabber - a simple client for the urlgrabber python package DESCRIPTION This is a thin client for the urlgrabber python package. It is provided mainly for helping debug the python package. It provides low-level access to most urlgrabber features from the shell. There are two types of options available for this program. They are 'client options' and 'module options'. Client options apply specifically to the behavior of this client, whereas module options are built-in options to the urlgrabber module. Both of these are avaible from the client command line, but they're documented a little differently. Client options are documented here, and module options are documented through the '--help <option>' syntax. CLIENT OPTIONS -h, --help In its short form, (-h) this provides a short usage discription. In its long form, it processes the remaining command line as help topics. Legal topics are: doc this info options a list of module options <option(s)> documentation for a given module option all documentation for all module options Examples: urlgrabber --help options urlgrabber --help copy_local -o FILE By default, downloaded data will be written to a file named using the basename of the url. For example, 'http://foo.com/index.html' will be written to 'index.html'. You can override this for your convenience or when necessary for urls like 'http://foo.com/' -O Print the local name of each downloaded file to STDOUT. This is helpful when not using the '-o' option, but is particularly useful when using --copy_local=0 (the default) on local files because local files will not be copied and the output filename will not be the same as that provided with '-o'. --repeat=N Grab each url N times. This is mostly for debugging keepalive. -p, --progress Use the default text-based progress meter. -v, --verbose=N Increment the verbosity level with each use of '-v' or set it directly with --verbose=N. Currently, distinct levels are 0-2. The default is 0. -d SPEC, --debug=SPEC Turn on internal urlgrabber debugging. This is equivalent (but overrides) running with the environment variable: URLGRABBER_DEBUG=SPEC SPEC can be of the form LEVEL,FILENAME, where LEVEL can be string (DEBUG, WARN, etc) or number. FILENAME can be the name of a file or "-" for STDOUT. The default is STDERR. Example: -d1,- logs everything to STDOUT. Note: this only works for python > 2.3 because it requires the logging package. -D A convenience alias for: --verbose=3 --progress --debug=INFO,- --profile Profile the actual fetching and print the results. """ MAINHELP = """usage: urlgrabber [options] <url> urlgrabber - a simple client for the urlgrabber python package options: -h, --help print this message --help doc print basic intro and documentation --help options list available options to the grabber module --help <option> print documentation for a module option --help all print documentation for all module options --<option>=VAL specify a module option. VAL must be a python value, including quotes in the case of strings. e.g. --user_agent='"foobar/2.0"' --output FILE -o FILE write output to FILE, otherwise the basename of the url will be used -O print the names of saved files to STDOUT --repeat=N grab each URL N times (mostly for debugging keepalive) -p, --progress use the default text progress meter -v increment the verbosity level (defaults to 0) --verbose=N set the verbosity level to N -d SPEC, --debug=SPEC turn on urlgrabber module debugging with SPEC=LEVEL,FILENAME. e.g. -d 1,debug.txt -D a convenience option equivalent to: --verbose=3 --progress --debug=INFO,- --profile profile the actual fetching and print the results """ # $Id: urlgrabber,v 1.7 2006/12/08 00:14:16 mstenner Exp $ import sys import getopt import re import urlgrabber.grabber from urlgrabber.grabber import URLGrabber, URLGrabberOptions, URLGrabError class client_options: def __init__(self): self.ug_options, self.ug_defaults = self.get_ug_options() self.process_command_line() def get_ug_options(self): ugo = URLGrabberOptions() ug_options = ['copy_local', 'keepalive', 'prefix', 'reget', 'data', 'quote', 'throttle', 'bandwidth', 'proxies', 'retry', 'retrycodes', 'range', 'user_agent', 'http_headers', 'ftp_headers', 'ssl_ca_cert', 'ssl_context', 'text', 'close_connection', 'cache_openers','timeout'] options_exclude = ['delegate', 'interrupt_callback', 'failure_callback', 'urlparser', 'opener', 'checkfunc', 'progress_obj'] for k in ugo.__dict__.keys(): if (k not in ug_options) and (k not in options_exclude): ug_options.append(k) #print k ug_defaults = {} for k in list(ug_options): try: ug_defaults[k] = repr(getattr(ugo, k)) except AttributeError: ug_options.remove(k) return ug_options, ug_defaults def process_command_line(self): short_options = 'vd:ho:OpD' long_options = ['profile', 'repeat=', 'verbose=', 'debug=', 'help', 'progress', 'output='] ug_long = [ o + '=' for o in self.ug_options ] try: optlist, args = getopt.getopt(sys.argv[1:], short_options, long_options + ug_long) except getopt.GetoptError, e: print >>sys.stderr, "Error:", e self.help([], ret=1) self.verbose = 0 self.debug = None self.outputfile = None self.localfile = 0 self.repeat = 1 self.progress = 0 self.profile = 0 self.ugops = {} self.args = args ug_dash = [ '--' + o for o in self.ug_options ] if not args: self.help(args) for (o, v) in optlist: if o == '--help' or o == '-h': self.help(args) if o == '--verbose': self.verbose = v if o == '-v': self.verbose += 1 if o == '-o': self.outputfile = v if o == '--output': self.outputfile = v if o == '-p' or o == '--progress': self.progress = 1 if o == '-d' or o == '--debug': self.debug = v if o == '--profile': self.profile = 1 if o == '-O': self.localfile = 1 if o == '--repeat': try: self.repeat = int(v) if self.repeat < 1: raise ValueError() except ValueError: print 'ERROR: repeat value must be an int >= 1' sys.exit(1) if o == '-D': self.verbose = 3 self.debug = "INFO,-" self.progress = 1 if o in ug_dash: try: val = eval(v) except Exception, e: print "error processing option value: %s" % v print e sys.exit(1) else: self.ugops[o[2:]] = val if len(self.args) > 1 and self.outputfile is not None: print "ERROR: cannot use -o when grabbing multiple files" sys.exit(1) def help(self, args, ret=0): if not args: print MAINHELP else: for a in args: m = getattr(self, 'help_'+a, None) if m is not None: m() elif a in self.ug_options: self.help_ug_option(a) else: print 'ERROR: no help on command "%s"' % a sys.exit(ret) def help_doc(self): print __doc__ def help_options(self): width = max(map(len, self.ug_options)) format = ' %-' + str(width) + 's = %s' hformat = ' %-' + str(width) + 's %s' print hformat % ('OPTION', 'DEFAULT') print '-'*(width + 20) for k in self.ug_options: print format % (k, self.ug_defaults[k]) def help_all(self): for k in self.ug_options: self.help_ug_option(k) def help_ug_option(self, option): help = '' m = re.search(r'^( '+option+'.*?)\s*^ {,2}\S', urlgrabber.grabber.__doc__, re.M|re.S) if m: print m.group(1) else: print ' %s: no help found for this option' % option print '' class ugclient: def __init__(self): op = client_options() self.op = op if op.verbose >= 2 and op.ugops: print "Module Options:" width = max(map(len, op.ugops.keys())) format = " %-" + str(width) + "s = %s" for k, v in op.ugops.items(): print format % (k, repr(v)) if op.debug: self.set_debug_logger(op.debug) if hasattr(urlgrabber.grabber, '_log_package_state'): urlgrabber.grabber._log_package_state() kwargs = dict(op.ugops) if op.progress: from urlgrabber.progress import text_progress_meter kwargs['progress_obj'] = text_progress_meter() self.g = URLGrabber(**kwargs) def run(self): for url in self.op.args: if self.op.verbose: print 'grabbing: %s' % url try: for i in range(0, self.op.repeat): f = self.g.urlgrab(url, self.op.outputfile) if self.op.localfile: print f except URLGrabError, e: print e sys.exit(1) def set_debug_logger(self, dbspec): try: dbinfo = dbspec.split(',') import logging level = logging._levelNames.get(dbinfo[0], None) if level is None: level = int(dbinfo[0]) if level < 1: raise ValueError() formatter = logging.Formatter('%(asctime)s %(message)s') if len(dbinfo) > 1: filename = dbinfo[1] else: filename = '' if filename == '': handler = logging.StreamHandler(sys.stderr) elif filename == '-': handler = logging.StreamHandler(sys.stdout) else: handler = logging.FileHandler(filename) handler.setFormatter(formatter) DBOBJ = logging.getLogger('urlgrabber') DBOBJ.addHandler(handler) DBOBJ.setLevel(level) except (KeyError, ImportError, ValueError): DBOBJ = None urlgrabber.grabber.set_logger(DBOBJ) if __name__ == '__main__': ugc = ugclient() if ugc.op.profile: import profile import pstats prof = profile.Profile() prof.run('ugc.run()') pstats.Stats(prof).strip_dirs().sort_stats('cumulative').print_stats() else: ugc.run()