mirror of
https://github.com/Sneed-Group/Poodletooth-iLand
synced 2024-12-26 21:22:27 -06:00
189 lines
6.3 KiB
Python
189 lines
6.3 KiB
Python
|
import os
|
||
|
|
from direct.directnotify.DirectNotifyGlobal import directNotify
|
||
|
|
from pandac.PandaModules import VirtualFileSystem
|
||
|
|
from pandac.PandaModules import Filename
|
||
|
|
from pandac.PandaModules import DSearchPath
|
||
|
|
from direct.showbase import ElementTree as ET
|
||
|
|
import LandingPageHTML
|
||
|
|
from StringIO import StringIO
|
||
|
|
|
||
|
|
class LandingPage:
|
||
|
|
notify = directNotify.newCategory("LandingPage")
|
||
|
|
def __init__(self):
|
||
|
|
self.headerTemplate = LandingPageHTML.header
|
||
|
|
self.footerTemplate = LandingPageHTML.footer
|
||
|
|
|
||
|
|
self.menu = {}
|
||
|
|
|
||
|
|
self.uriToTitle = {}
|
||
|
|
|
||
|
|
self.quickStats = [[],{}]
|
||
|
|
|
||
|
|
self.addQuickStat("Pages Served", 0, 0)
|
||
|
|
|
||
|
|
self.favicon = self.readFavIcon()
|
||
|
|
|
||
|
|
|
||
|
|
def readFavIcon(self):
|
||
|
|
vfs = VirtualFileSystem.getGlobalPtr()
|
||
|
|
filename = Filename('favicon.ico')
|
||
|
|
|
||
|
|
searchPath = DSearchPath()
|
||
|
|
searchPath.appendDirectory(Filename('.'))
|
||
|
|
searchPath.appendDirectory(Filename('etc'))
|
||
|
|
searchPath.appendDirectory(Filename.fromOsSpecific(os.path.expandvars('$DIRECT/src/http')))
|
||
|
|
searchPath.appendDirectory(Filename.fromOsSpecific(os.path.expandvars('direct/src/http')))
|
||
|
|
searchPath.appendDirectory(Filename.fromOsSpecific(os.path.expandvars('direct/http')))
|
||
|
|
found = vfs.resolveFilename(filename,searchPath)
|
||
|
|
if not found:
|
||
|
|
raise "Couldn't find direct/http/favicon.ico"
|
||
|
|
|
||
|
|
return vfs.readFile(filename, 1)
|
||
|
|
|
||
|
|
|
||
|
|
def addTab(self, title, uri):
|
||
|
|
self.menu[title] = uri
|
||
|
|
self.uriToTitle[uri] = title
|
||
|
|
|
||
|
|
def getMenu(self, activeTab):
|
||
|
|
return LandingPageHTML.getTabs(self.menu,activeTab)
|
||
|
|
|
||
|
|
def getMenuTags(self, activeTab):
|
||
|
|
return LandingPageHTML.getTabTags(self.menu, activeTab)
|
||
|
|
|
||
|
|
def getHeader(self, activeTab = "Main", headTag=None, bodyTag=None):
|
||
|
|
if headTag is None:
|
||
|
|
headTag = ET.Element('head')
|
||
|
|
if bodyTag is None:
|
||
|
|
bodyTag = ET.Element('body')
|
||
|
|
|
||
|
|
# make sure each component has elements to make formatting consistent
|
||
|
|
headTag.append(ET.Comment(''))
|
||
|
|
bodyTag.append(ET.Comment(''))
|
||
|
|
|
||
|
|
fileStr = StringIO()
|
||
|
|
ET.ElementTree(headTag).write(fileStr, encoding='utf-8')
|
||
|
|
headTagStr = unicodeUtf8(fileStr.getvalue())
|
||
|
|
# remove the tag closer
|
||
|
|
# </head>
|
||
|
|
headTagStr = headTagStr[:headTagStr.rindex('<')]
|
||
|
|
|
||
|
|
# fill in the prefab body tag content
|
||
|
|
titleStr = LandingPageHTML.title
|
||
|
|
landing = ET.Element('body')
|
||
|
|
LandingPageHTML.addBodyHeaderAndContent(landing, titleStr, self.getMenuTags(activeTab))
|
||
|
|
|
||
|
|
fileStr = StringIO()
|
||
|
|
ET.ElementTree(landing).write(fileStr, encoding='utf-8')
|
||
|
|
landingStr = unicodeUtf8(fileStr.getvalue())
|
||
|
|
# remove <body>
|
||
|
|
landingStr = landingStr[landingStr.index('>')+1:]
|
||
|
|
# remove tag closers
|
||
|
|
for i in xrange(3):
|
||
|
|
# </body>
|
||
|
|
# contents </div>
|
||
|
|
# </center>
|
||
|
|
landingStr = landingStr[:landingStr.rindex('<')]
|
||
|
|
|
||
|
|
fileStr = StringIO()
|
||
|
|
ET.ElementTree(bodyTag).write(fileStr, encoding='utf-8')
|
||
|
|
bodyTagStr = unicodeUtf8(fileStr.getvalue())
|
||
|
|
# extract <body>
|
||
|
|
bodyStr = bodyTagStr[bodyTagStr.index('>')+1:]
|
||
|
|
bodyTagStr = bodyTagStr[:bodyTagStr.index('>')+1]
|
||
|
|
|
||
|
|
bodyStr = bodyTagStr + '\n' + landingStr + '\n' + bodyStr
|
||
|
|
|
||
|
|
s = self.headerTemplate % {'titlestring': titleStr,
|
||
|
|
'headTag' : headTagStr,
|
||
|
|
'bodyTag': bodyStr,
|
||
|
|
}
|
||
|
|
return s
|
||
|
|
|
||
|
|
def getFooter(self):
|
||
|
|
return self.footerTemplate % {'contact' : LandingPageHTML.contactInfo}
|
||
|
|
|
||
|
|
def getServicesPage(self, uriToHandler):
|
||
|
|
output = ""
|
||
|
|
|
||
|
|
uriList = uriToHandler.keys()
|
||
|
|
|
||
|
|
uriList.sort()
|
||
|
|
|
||
|
|
autoList = []
|
||
|
|
|
||
|
|
if "/" in uriList:
|
||
|
|
uriList.remove("/")
|
||
|
|
autoList.append("/")
|
||
|
|
if "/services" in uriList:
|
||
|
|
uriList.remove("/services")
|
||
|
|
autoList.append("/services")
|
||
|
|
if "/default.css" in uriList:
|
||
|
|
uriList.remove("/default.css")
|
||
|
|
autoList.append("/default.css")
|
||
|
|
if "/favicon.ico" in uriList:
|
||
|
|
uriList.remove("/favicon.ico")
|
||
|
|
autoList.append("/favicon.ico")
|
||
|
|
|
||
|
|
output += LandingPageHTML.getURITable(title="Application",uriList=uriList,uriToHandler=uriToHandler)
|
||
|
|
|
||
|
|
output += LandingPageHTML.getURITable(title="Admin",uriList=autoList,uriToHandler=uriToHandler)
|
||
|
|
|
||
|
|
return output
|
||
|
|
|
||
|
|
def populateMainPage(self, body):
|
||
|
|
LandingPageHTML.mainPageBody = body
|
||
|
|
|
||
|
|
def setTitle(self, title):
|
||
|
|
if LandingPageHTML.title == LandingPageHTML.defaultTitle:
|
||
|
|
LandingPageHTML.title = title
|
||
|
|
else:
|
||
|
|
LandingPageHTML.title = LandingPageHTML.title + " + " + title
|
||
|
|
|
||
|
|
def setDescription(self,desc):
|
||
|
|
if LandingPageHTML.description == LandingPageHTML.defaultDesc:
|
||
|
|
LandingPageHTML.description = desc
|
||
|
|
else:
|
||
|
|
LandingPageHTML.description = LandingPageHTML.description + "</P>\n<P>" + desc
|
||
|
|
|
||
|
|
def setContactInfo(self,info):
|
||
|
|
LandingPageHTML.contactInfo = info
|
||
|
|
|
||
|
|
def getDescription(self):
|
||
|
|
return LandingPageHTML.description
|
||
|
|
|
||
|
|
def getQuickStatsTable(self):
|
||
|
|
return LandingPageHTML.getQuickStatsTable(self.quickStats)
|
||
|
|
|
||
|
|
def getMainPage(self):
|
||
|
|
return LandingPageHTML.mainPageBody % {"description" : self.getDescription(),
|
||
|
|
"quickstats" : self.getQuickStatsTable()}
|
||
|
|
|
||
|
|
def getStyleSheet(self):
|
||
|
|
return LandingPageHTML.stylesheet
|
||
|
|
|
||
|
|
def getFavIcon(self):
|
||
|
|
return self.favicon
|
||
|
|
|
||
|
|
def skin(self, body, uri, headTag=None, bodyTag=None):
|
||
|
|
title = self.uriToTitle.get(uri,"Services")
|
||
|
|
return self.getHeader(title, headTag, bodyTag) + body + self.getFooter()
|
||
|
|
|
||
|
|
def addQuickStat(self,item,value,position):
|
||
|
|
if item in self.quickStats[1]:
|
||
|
|
assert self.notify.warning("Ignoring duplicate addition of quickstat %s." % item)
|
||
|
|
return
|
||
|
|
|
||
|
|
self.quickStats[0].insert(position,item)
|
||
|
|
self.quickStats[1][item] = value
|
||
|
|
|
||
|
|
def updateQuickStat(self,item,value):
|
||
|
|
assert item in self.quickStats[1]
|
||
|
|
|
||
|
|
self.quickStats[1][item] = value
|
||
|
|
|
||
|
|
def incrementQuickStat(self,item):
|
||
|
|
assert item in self.quickStats[1]
|
||
|
|
|
||
|
|
self.quickStats[1][item] += 1
|