mirror of
https://github.com/encounter/phantomjs.git
synced 2026-03-30 11:35:11 -07:00
IceArmy
65563ca89d
* Convert all QString and QVariant types * Replaced all references to QString with Python str object - Fixed bug where phantom.version didn't work - Don't cd to scripts directory anymore (but loadScript still loads scripts from directory relative to it) - clipRect.setter and viewportSize.setter both have improved system now. * if you miss or neglect to specify an attritube (width, height, etc), there will no longer be an exception. it will be handled gracefully, and whatever you didn't specify won't be changed * if you enter a negative value on any of them, it'll be reset to 0 (except for phantom.clipRect top and left) and you won't loose your attempted changes - Cleaned up Webpage's javaScriptConsoleMessage print JavaScript (sourceID) - Allow viewportSize to be set to 0 (default) - General cleanup
137 lines
4.6 KiB
Python
137 lines
4.6 KiB
Python
#!/usr/bin/env python
|
|
'''
|
|
This file is part of the PyPhantomJS project.
|
|
|
|
Copyright (C) 2011 James Roe <roejames12@hotmail.com>
|
|
Copyright (C) 2010-2011 Ariya Hidayat <ariya.hidayat@gmail.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/>.
|
|
'''
|
|
|
|
# automatically convert Qt types by using api 2
|
|
import sip
|
|
sip.setapi('QString', 2)
|
|
sip.setapi('QVariant', 2)
|
|
|
|
import os, sys, resources
|
|
|
|
from phantom import Phantom
|
|
from utils import argParser, MessageHandler, version
|
|
|
|
from PyQt4.QtCore import qInstallMsgHandler
|
|
from PyQt4.QtGui import QIcon, QApplication
|
|
|
|
# make keyboard interrupt quit program
|
|
import signal
|
|
signal.signal(signal.SIGINT, signal.SIG_DFL)
|
|
|
|
def parseArgs(args):
|
|
# Handle all command-line options
|
|
p = argParser()
|
|
arg_data = p.parse_known_args(args)
|
|
args = arg_data[0]
|
|
args.script_args = arg_data[1]
|
|
|
|
if args.upload_file:
|
|
# process the tags
|
|
item_buffer = {}
|
|
for i in range(len(args.upload_file)):
|
|
item = args.upload_file[i].split('=')
|
|
if len(item) < 2 or not len(item[1]):
|
|
# if buffer is empty, or tag has no
|
|
# value 'tag=', print help and exit
|
|
if not len(item_buffer) or \
|
|
item[1:] and not item[1:][0]:
|
|
p.print_help()
|
|
sys.exit(1)
|
|
|
|
# this is a bug workaround for argparse.
|
|
# if you call parse_known_args, and you
|
|
# have an --option script arg, the args
|
|
# get jumbled up, and it's inconsistent
|
|
#
|
|
# we're just going to check for -- and
|
|
# swap it all back to the right order
|
|
if args.script_args:
|
|
for i in range(len(args.upload_file)):
|
|
if not args.upload_file[i].count('='):
|
|
# insert the arg after --option (make sure it's not None)
|
|
if args.script:
|
|
args.script_args.insert(1, args.script)
|
|
# insert value args before --option
|
|
if args.upload_file[i+1:]:
|
|
arg_buffer = args.upload_file[i+1:]
|
|
arg_buffer.reverse()
|
|
for val in arg_buffer:
|
|
args.script_args.insert(0, val)
|
|
args.script = args.upload_file[i]
|
|
break
|
|
else:
|
|
args.script = args.upload_file[i]
|
|
args.script_args = args.upload_file[i+1:]
|
|
break
|
|
|
|
# duplicate tag checking
|
|
if item[0] in item_buffer:
|
|
sys.exit('Multiple tags named \'%s\' were found' % item[0])
|
|
|
|
item_buffer[item[0]] = item[1]
|
|
|
|
# make sure files exist
|
|
for tag in item_buffer:
|
|
if not os.path.exists(item_buffer[tag]):
|
|
sys.exit('No such file or directory: \'%s\'' % item_buffer[tag])
|
|
args.upload_file = item_buffer
|
|
|
|
if args.proxy:
|
|
item = args.proxy.split(':')
|
|
if len(item) < 2 or not len(item[1]):
|
|
p.print_help()
|
|
sys.exit(1)
|
|
args.proxy = item
|
|
|
|
if not args.script:
|
|
p.print_help()
|
|
sys.exit(1)
|
|
|
|
try:
|
|
args.script = open(args.script)
|
|
except IOError as (errno, stderr):
|
|
sys.exit('%s: \'%s\'' % (stderr, args.script))
|
|
|
|
return args
|
|
|
|
def main():
|
|
args = parseArgs(sys.argv[1:])
|
|
|
|
# register an alternative Message Handler
|
|
messageHandler = MessageHandler(args.verbose)
|
|
qInstallMsgHandler(messageHandler.process)
|
|
|
|
app = QApplication(sys.argv)
|
|
|
|
app.setWindowIcon(QIcon(':/resources/pyphantomjs-icon.png'))
|
|
app.setApplicationName('PyPhantomJS')
|
|
app.setOrganizationName('Umaclan Development')
|
|
app.setOrganizationDomain('www.umaclan.com')
|
|
app.setApplicationVersion(version)
|
|
|
|
phantom = Phantom(args, app)
|
|
phantom.execute()
|
|
app.exec_()
|
|
sys.exit(phantom.returnValue())
|
|
|
|
if __name__ == '__main__':
|
|
main()
|