1

Adding Ajax code, bumping version number

This commit is contained in:
Alex Yatskov 2016-05-01 17:30:08 -07:00
parent 9f9d2221db
commit 820348f7c2
3 changed files with 169 additions and 1 deletions

167
yomi_base/ajax.py Normal file
View File

@ -0,0 +1,167 @@
# -*- coding: utf-8 -*-
# Copyright (C) 2016 Alex Yatskov <alex@foosoft.net>
# Author: Alex Yatskov <alex@foosoft.net>
#
# 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 json
import select
import socket
class AjaxRequest:
def __init__(self, headers, body):
self.headers = headers
self.body = body
class AjaxClient:
def __init__(self, sock, callback):
self.sock = sock
self.callback = callback
self.readBuff = ''
self.readBuff = ''
def advance(self, recvSize=1):
if self.sock is None:
return False
rlist, wlist = select.select([self.sock], [self.sock], [], 0)[:2]
if rlist:
msg = self.sock.recv(recvSize)
if not msg:
self.close()
return False
self.readBuff += msg
req, length = self.parseRequest(self.readBuff)
if req is not None:
self.readBuff = self.readBuff[length:]
self.readBuff += self.callback(req)
if wlist and self.readBuff:
length = self.sock.send(self.readBuff)
self.readBuff = self.readBuff[length:]
if not self.readBuff:
self.close()
return False
return True
def close(self):
if self.sock is not None:
self.sock.close()
self.sock = None
self.readBuff = ''
self.readBuff = ''
def parseRequest(self, data):
parts = data.split('\r\n\r\n', 1)
if len(parts) == 1:
return None, 0
headers = {}
for line in parts[0].split('\r\n'):
pair = line.split(': ')
headers[pair[0]] = pair[1] if len(pair) > 1 else None
headerLength = len(parts[0]) + 4
bodyLength = int(headers['Content-Length'])
totalLength = headerLength + bodyLength
if totalLength > len(data):
return None, 0
body = data[headerLength : totalLength]
return AjaxRequest(headers, body), totalLength
class AjaxServer:
def __init__(self, callback):
self.callback = callback
self.clients = []
self.sock = None
def advance(self):
if self.sock is not None:
self.acceptClients()
self.advanceClients()
def acceptClients(self):
rlist = select.select([self.sock], [], [], 0)[0]
if not rlist:
return
clientSock = self.sock.accept()[0]
if clientSock is not None:
clientSock.setblocking(False)
self.clients.append(AjaxClient(clientSock, self.callbackWrapper))
def advanceClients(self):
self.clients = filter(lambda c: c.advance(), self.clients)
def listen(self, address='127.0.0.1', port=8888, backlog=5):
self.close()
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self.sock.setblocking(False)
self.sock.bind((address, port))
self.sock.listen(backlog)
def callbackWrapper(self, req):
body = json.dumps(self.callback(json.loads(req.body)))
resp = ''
headers = {
'HTTP/1.1 200 OK': None,
'Content-Type': 'application/json',
'Content-Length': str(len(body)),
'Access-Control-Allow-Origin': '*'
}
for key, value in headers.items():
if value is None:
resp += '{}\r\n'.format(key)
else:
resp += '{}: {}\r\n'.format(key, value)
resp += '\r\n'
resp += body
return resp
def close(self):
if self.sock is not None:
self.sock.close()
self.sock = None
for client in self.clients:
client.close()
self.clients = []

View File

@ -17,5 +17,5 @@
c = { c = {
'appVersion': '0.17b', 'appVersion': '0.20b',
} }

View File

@ -115,6 +115,7 @@ class MainWindowReader(QtGui.QMainWindow, gen.reader_ui.Ui_MainWindowReader):
self.onActionPreferences() self.onActionPreferences()
def applyPreferencesContent(self): def applyPreferencesContent(self):
palette = self.textContent.palette() palette = self.textContent.palette()
palette.setColor(QtGui.QPalette.Base, QtGui.QColor(self.preferences['bgColor'])) palette.setColor(QtGui.QPalette.Base, QtGui.QColor(self.preferences['bgColor']))