mirror of
https://github.com/encounter/phantomjs.git
synced 2026-03-30 11:35:11 -07:00
IceArmy
a313d1befd
Refactored most of functionalities to WebPage object WebPage object can have settings Support different HTTP operations for loading a page Simplify buffer handling in render() Refactor render(), split the buffer rendering to its own function Render the web page using the tiling approach WebPage callback for it's console message Make exit() synchronous WebPage callback for its JS alert Updated all examples
130 lines
4.4 KiB
Python
130 lines
4.4 KiB
Python
'''
|
|
This file is part of the PyPhantomJS project.
|
|
|
|
Copyright (C) 2011 James Roe <roejames12@hotmail.com>
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program 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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
'''
|
|
|
|
import argparse, sys
|
|
|
|
from plugincontroller import Bunch, do_action
|
|
|
|
from PyQt4.QtCore import QDateTime, Qt, QtDebugMsg, QtWarningMsg, QtCriticalMsg, QtFatalMsg
|
|
|
|
version_major, version_minor, version_patch = (1, 2, 0)
|
|
version = '%d.%d.%d' % (version_major, version_minor, version_patch)
|
|
|
|
license = '''
|
|
PyPhantomJS Version %s
|
|
|
|
Copyright (C) 2011 James Roe <roejames12@hotmail.com>
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program 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 General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
''' % version
|
|
|
|
|
|
def argParser():
|
|
parser = argparse.ArgumentParser(
|
|
description='Minimalistic headless WebKit-based JavaScript-driven tool',
|
|
usage='%(prog)s [options] script.[js|coffee] [script argument [script argument ...]]',
|
|
formatter_class=argparse.RawTextHelpFormatter
|
|
)
|
|
|
|
parser.add_argument('script', metavar='script.[js|coffee]', nargs='?',
|
|
help='The script to execute, and any args to pass to it'
|
|
)
|
|
|
|
parser.add_argument('--disk-cache', default='no',
|
|
choices=['yes', 'no'],
|
|
help='Enable disk cache (default: %(default)s)'
|
|
)
|
|
parser.add_argument('--ignore-ssl-errors', default='no',
|
|
choices=['yes', 'no'],
|
|
help='Ignore SSL errors (default: %(default)s)'
|
|
)
|
|
parser.add_argument('--load-images', default='yes',
|
|
choices=['yes', 'no'],
|
|
help='Load all inlined images (default: %(default)s)'
|
|
)
|
|
parser.add_argument('--load-plugins', default='no',
|
|
choices=['yes', 'no'],
|
|
help='Load all plugins (i.e. Flash, Silverlight, ...) (default: %(default)s)'
|
|
)
|
|
parser.add_argument('--proxy', metavar='address:port',
|
|
help='Set the network proxy'
|
|
)
|
|
parser.add_argument('--upload-file', nargs='*',
|
|
metavar='tag=file', help='Upload 1 or more files'
|
|
)
|
|
parser.add_argument('-v', '--verbose', action='store_true',
|
|
help='Show verbose debug messages'
|
|
)
|
|
parser.add_argument('--version',
|
|
action='version', version=license,
|
|
help='show this program\'s version and license'
|
|
)
|
|
|
|
do_action('ArgParser', Bunch(locals()))
|
|
|
|
return parser
|
|
|
|
|
|
class MessageHandler:
|
|
def __init__(self, verbose):
|
|
self.verbose = verbose
|
|
|
|
def process(self, msgType, msg):
|
|
now = QDateTime.currentDateTime().toString(Qt.ISODate)
|
|
|
|
if msgType == QtDebugMsg:
|
|
if self.verbose:
|
|
print '%s [DEBUG] %s' % (now, msg)
|
|
elif msgType == QtWarningMsg:
|
|
print >> sys.stderr, '%s [WARNING] %s' % (now, msg)
|
|
elif msgType == QtCriticalMsg:
|
|
print >> sys.stderr, '%s [CRITICAL] %s' % (now, msg)
|
|
elif msgType == QtFatalMsg:
|
|
print >> sys.stderr, '%s [FATAL] %s' % (now, msg)
|
|
|
|
|
|
class SafeStreamFilter(object):
|
|
'''Convert string to something safe'''
|
|
def __init__(self, target):
|
|
self.target = target
|
|
self.encoding = 'utf-8'
|
|
self.errors = 'replace'
|
|
self.encode_to = self.target.encoding
|
|
|
|
def write(self, s):
|
|
s = self.encode(s)
|
|
self.target.write(s)
|
|
|
|
def flush(self):
|
|
self.target.flush()
|
|
|
|
def encode(self, s):
|
|
return s.encode(self.encode_to, self.errors).decode(self.encode_to)
|