162 lines
6.2 KiB
Python
162 lines
6.2 KiB
Python
"""
|
|
Qt WebKit Browser for local access to internal Django Views.
|
|
"""
|
|
import os, logging
|
|
from PyQt4 import QtCore, QtGui, QtWebKit, QtNetwork
|
|
from django.test import Client
|
|
|
|
class DebugPage(QtWebKit.QWebPage):
|
|
def sayMyName(self):
|
|
return 'DebugPage'
|
|
|
|
class DejaWebView(QtWebKit.QWebView):
|
|
'''
|
|
BaseDir. Get rid of it?
|
|
'''
|
|
def __init__(self, *args, **kwargs):
|
|
basedir = kwargs.pop('basedir', None)
|
|
QtWebKit.QWebView.__init__(self, *args, **kwargs)
|
|
oldManager = self.page().networkAccessManager()
|
|
self.setPage(DebugPage())
|
|
self.page().setNetworkAccessManager(DejaNetworkAccessManager(self, basedir))
|
|
|
|
def set_basedir(self, basedir):
|
|
self.page().setNetworkAccessManager(DejaNetworkAccessManager(self, basedir))
|
|
|
|
class DejaNetworkAccessManager(QtNetwork.QNetworkAccessManager):
|
|
'''
|
|
The Deja Network Access Manager provides access to two new protocols:
|
|
- page:/// tries to resolve a page internally via a django test client.
|
|
- res:/// direct access to a resource.
|
|
|
|
USE_NETWORK delegates to other network access manager protocols, if False, it will not
|
|
allow any requests outside these two protocols
|
|
(hopefully disabling network access for your internal browser)
|
|
|
|
Note, if page does not find the page, a res:/// attempt is made automatically.
|
|
This has to be expanded!
|
|
|
|
Note2: not sure if cookies and sessions will work this way!
|
|
'''
|
|
USE_NETWORK = False
|
|
def __init__(self, parent=None, basedir=None):
|
|
QtNetwork.QNetworkAccessManager.__init__(self, parent=None)
|
|
if not basedir:
|
|
# take current dir as basedir.
|
|
self.basedir = os.path.dirname(os.path.abspath(__file__))
|
|
else:
|
|
self.basedir = basedir
|
|
|
|
def createRequest(self, operation, request, data):
|
|
scheme = request.url().scheme()
|
|
if scheme != 'page' and scheme != 'res':
|
|
if self.USE_NETWORK:
|
|
return QtNetwork.QNetworkAccessManager.createRequest(self, operation, request, data)
|
|
elif scheme == 'page':
|
|
if operation == self.GetOperation:
|
|
# Handle page:// URLs separately by creating custom
|
|
# QNetworkReply objects.
|
|
reply = PageReply(self, request.url(), self.GetOperation)
|
|
#print('here')
|
|
#print reply
|
|
return reply
|
|
elif operation == self.PostOperation:
|
|
#print data.readAll()
|
|
#print request
|
|
reply = PageReply(self, request.url(), self.PostOperation)
|
|
return reply
|
|
elif scheme == 'res':
|
|
if operation == self.GetOperation:
|
|
return ImageReply(self, request.url(), self.GetOperation, self.basedir)
|
|
else:
|
|
if self.USE_NETWORK:
|
|
return QtNetwork.QNetworkAccessManager.createRequest(self, operation, request, data)
|
|
return NoNetworkReply(self, request.url(), self.GetOperation)
|
|
|
|
class BasePageReply(QtNetwork.QNetworkReply):
|
|
content_type = 'text/html; charset=utf-8'
|
|
def __init__(self, parent, url, operation):
|
|
QtNetwork.QNetworkReply.__init__(self, parent)
|
|
self.content = self.initialize_content(url, operation)
|
|
self.offset = 0
|
|
self.setHeader(QtNetwork.QNetworkRequest.ContentTypeHeader, self.get_content_type())
|
|
self.setHeader(QtNetwork.QNetworkRequest.ContentLengthHeader, len(self.content))
|
|
QtCore.QTimer.singleShot(0, self, QtCore.SIGNAL('readyRead()'))
|
|
QtCore.QTimer.singleShot(0, self, QtCore.SIGNAL('finished()'))
|
|
self.open(self.ReadOnly | self.Unbuffered)
|
|
self.setUrl(url)
|
|
|
|
def get_content_type(self):
|
|
return self.content_type
|
|
|
|
def initialize_content(self, url, operation):
|
|
return '''
|
|
<html>
|
|
<head><title>Empty Page</title></head>
|
|
<body>This is an empty page. If you see this, you need to subclass BasePageReply.</body>
|
|
</html>
|
|
'''
|
|
|
|
def abort(self):
|
|
pass
|
|
|
|
def bytesAvailable(self):
|
|
return len(self.content) - self.offset + QtNetwork.QNetworkReply.bytesAvailable(self)
|
|
|
|
def isSequential(self):
|
|
return True
|
|
|
|
def readData(self, maxSize):
|
|
if self.offset < len(self.content):
|
|
end = min(self.offset + maxSize, len(self.content))
|
|
data = self.content[self.offset:end]
|
|
self.offset = end
|
|
return data
|
|
|
|
class PageReply(BasePageReply):
|
|
def initialize_content(self, url, operation):
|
|
c = Client()
|
|
print "Response for %s, method %s" % (url.path(), operation)
|
|
if operation == DejaNetworkAccessManager.GetOperation:
|
|
response = c.get(unicode(url.path()), )
|
|
elif operation == DejaNetworkAccessManager.PostOperation:
|
|
response = c.post(unicode(url.path()))
|
|
# response code
|
|
print "Response Status: %s" % response.status_code
|
|
# note: on a 404, we might need to trigger file response.
|
|
return response.content
|
|
|
|
class NoNetworkReply(BasePageReply):
|
|
def initialize_content(self, url, operation):
|
|
return '''
|
|
<html>
|
|
<head><title>No Network Access.</title></head>
|
|
<body>
|
|
Internal access to the network has been disabled.
|
|
</body>
|
|
</html>
|
|
'''
|
|
|
|
class ResourceReply(BasePageReply):
|
|
content_type = 'image/png'
|
|
|
|
def __init__(self, parent, url, operation, basedir):
|
|
self.basedir = basedir
|
|
BasePageReply.__init__(self, parent, url, operation)
|
|
|
|
def determine_content_type(self, path):
|
|
return self.content_type
|
|
|
|
def initialize_content(self, url, operation):
|
|
path = os.path.join(self.basedir, unicode(url.path()).lstrip('/'))
|
|
if not os.path.exists(path):
|
|
logging.error('Image does not exist: %s' % path)
|
|
return ''
|
|
#h = url.host()
|
|
# @todo: host checks?
|
|
try:
|
|
f = open(path, 'rb')
|
|
return f.read()
|
|
finally:
|
|
f.close()
|
|
|