def renderCSV(queryResults) :
import csv
import StringIO
csvOutput = StringIO.StringIO()
writer = csv.writer(csvOutput)
# Get heading list.
fieldsInfo = queryResults.getFieldsInfo()
displayHeadings = []
for nField in fieldsInfo :
displayHeadings.append(str(fieldsInfo[nField].fieldHeading))
writer.writerow(displayHeadings)
writer.writerow(["" for i in range(0, len(displayHeadings))])
# Get data matrix
for queryItem in queryResults.queryResults :
dataRow = []
for nField in fieldsInfo :
field, dataPage, fieldHeading = queryItem.getData(nField)
displayData = datatype.convert(datatype.convert(field.data, mode=datatype.TO_PYTHON), mode=datatype.TO_STRING, form=datatype.STRING_FORM_SHORT)
dataRow.append(displayData)
writer.writerow(dataRow)
return csvOutput.getvalue()
def flatten_data(self) :
debugOutput("Flattening data.")
data = {}
wikidpageFields = self.wikidpage.getFields().getDict()
for nFieldName in wikidpageFields :
field = wikidpageFields[nFieldName]
if not wikidbase.core.pagecontent.isDataField(field):
continue
debugOutput("field.data %s" % field.data)
dataObject = datatype.convert(field.data, mode=datatype.TO_PYTHON)
# Expand list data to a list.
if dataObject and type(dataObject) == str and self.wikidpage.context :
nContext = wikidbase.core.context.normaliseTerm(self.wikidpage.context)
nFieldName = field.nName
contextFieldState = wikidbase.core.state.getContextFieldState(nContext, nFieldName)
try :
widgetType = contextFieldState[wikidbase.core.context.LIST_STYLE]
except :
widgetType = wikidbase.core.context.AUTO_WIDGET
if widgetType == wikidbase.core.context.MS_LIST :
data[field.name] = dataObject.splitlines()
else :
data[field.name] = dataObject
else :
data[field.name] = dataObject
debugOutput("data %s" % data)
return data
def __str__(self):
"""A string representation of this object, for debugging."""
displayString = ""
firstDisplayed = False
for attribute in self.getOrderedAttributes() :
if attribute.type != WP_FIELD :
continue
if attribute.data and type(datatype.convert(attribute.data, mode=datatype.TO_PYTHON)) == str and "\n" not in attribute.data:
displayString += "%s: %s, " % (attribute.name, attribute.data)
firstDisplayed = True
return displayString.rstrip(", ")
def __str__(self):
"""A string representation of this object, for debugging."""
displayString = ""
firstDisplayed = False
for attribute in self.getOrderedAttributes():
if attribute.type != WP_FIELD:
continue
if attribute.data and type(
datatype.convert(attribute.data, mode=datatype.TO_PYTHON)
) == str and "\n" not in attribute.data:
displayString += "%s: %s, " % (attribute.name, attribute.data)
firstDisplayed = True
return displayString.rstrip(", ")
def render(self, data):
if data is None:
data = ''
maxlength = ''
if self.maxlength:
maxlength = 'maxlength="%s" ' % self.maxlength
if isinstance(data, unicode):
data = data.encode(django.conf.settings.DEFAULT_CHARSET)
triggerID = self.get_id()+"_trigger"
return """<input type="%s" id="%s" class="v%s%s" name="%s" size="%s" value="%s" %s/> <button type="reset" id="%s">...</button>%s""" % \
(self.input_type, self.get_id(), self.__class__.__name__, self.is_required and ' required' or '',
self.field_name, self.length, escape(datatype.convert(data, mode=datatype.TO_STRING, form=datatype.STRING_FORM_SHORT)), maxlength, triggerID, self._calendarSetupString(self.get_id(), triggerID))
def render(self, data):
if data is None:
data = ''
maxlength = ''
if self.maxlength:
maxlength = 'maxlength="%s" ' % self.maxlength
if isinstance(data, unicode):
data = data.encode(django.conf.settings.DEFAULT_CHARSET)
triggerID = self.get_id() + "_trigger"
return """<input type="%s" id="%s" class="v%s%s" name="%s" size="%s" value="%s" %s/> <button type="reset" id="%s">...</button>%s""" % \
(self.input_type, self.get_id(), self.__class__.__name__, self.is_required and ' required' or '',
self.field_name, self.length, escape(datatype.convert(data, mode=datatype.TO_STRING, form=datatype.STRING_FORM_SHORT)), maxlength, triggerID, self._calendarSetupString(self.get_id(), triggerID))
def _selectWidget(self, field, wikidpage) :
"""Select the appropriate widget for a wikidpage field."""
data = field.data
# Try to get the context for this page
wikidpageContext = None
if wikidpage.context :
wikidpageContexts = wikidbase.core.context.getContexts()
nContext = wikidbase.core.normaliseTerm(wikidpage.context)
if nContext in wikidpageContexts :
wikidpageContext = wikidpageContexts[nContext]
# If there is no data to guess the widget from, look in the context of this page.
if not data and wikidpageContext :
if field.nName in wikidpageContext.dataFieldContexts :
try :
data = wikidpageContext.dataFieldContexts[field.nName].representativeData.stack[0]
except :
pass
dataObject = datatype.convert(data, mode=datatype.TO_PYTHON)
debugOutput("field %s, data: %s, dataObject: %s, type: %s" % (field.nName, data, dataObject, type(dataObject)))
formField = None
# TODO:
#
# if date - > date
# if bdolean -> checkbox
# if multiline data
# if not list -> text area
# else -> multiselect
# if date - > date
# if boolean -> checkbox
# else (if just single line text)
# if mlist -> multiselect
# if slist -> select
# else :
# single text field.
# Load the context state.
widgetType = wikidbase.core.context.AUTO_WIDGET
listSort = wikidbase.core.context.SORT_MOST_COMMON
listChoices = []
if wikidpage.context :
nContext = wikidbase.core.context.normaliseTerm(wikidpage.context)
nFieldName = field.nName
contextFieldState = wikidbase.core.state.getContextFieldState(nContext, nFieldName)
try :
widgetType = contextFieldState[wikidbase.core.context.LIST_STYLE]
listSort = contextFieldState[wikidbase.core.context.LIST_SORT]
listChoices = contextFieldState[wikidbase.core.context.LIST_CHOICES].splitlines()
except :
pass
debugOutput("widgetType %s, listSort %s, listChoices %s" % (widgetType, listSort, listChoices))
# Compute selection choices.
if wikidpageContext and field.nName in wikidpageContext.dataFieldContexts :
choices = [choice for choice in wikidpageContext.dataFieldContexts[field.nName].selectionSet]
else :
choices = []
# Append field state choices
choices[0:0] = listChoices
if data :
# Note, we interpret lines as multiple selections.
dataLines = data.splitlines()
for dataItem in dataLines :
if dataItem not in choices :
choices[0:0] = [dataItem]
# Remove duplicates - could be more efficient.
uniqueChoices = []
for choice in choices :
if choice not in uniqueChoices :
uniqueChoices.append(choice)
choices = uniqueChoices
# Sort choices
if listSort == wikidbase.core.context.SORT_ASCEND :
choices.sort()
elif listSort == wikidbase.core.context.SORT_DESCEND :
choices.sort(reverse=True)
choices = [[choice, choice] for choice in choices]
# TODO: We should be able to add hooks here so people can add widgets .
# In wikidbase, there are two ways to choose a widget for a field:
# * Explicit: the user sets the widget to single-select/multi-select list, textarea, etc.
# * Implicit: the widget is chosen based on data or on context data for that field if empty (e.g. date string -> cal widget).
if widgetType == wikidbase.core.context.TEXTAREA :
formField = widgets.LargeTextFieldExtra(field_name=field.name, is_required=False, rows=6,cols=40)
# TODO: Should we use WP_MULTILINE_FIELD to choose widget now?
elif data and (field.type == wikidbase.core.pagecontent.WP_MULTILINE_FIELD or "\n" in data) :
if widgetType == wikidbase.core.context.AUTO_WIDGET :
formField = widgets.LargeTextFieldExtra(field_name=field.name, is_required=False, rows=6,cols=40)
else :
formField = forms.SelectMultipleField(field_name=field.name, is_required=False, choices=choices)
elif type(dataObject) == datetime.date :
formField = widgets.NickDatetimeField(field_name=field.name, is_required=False, showsTime=False)
elif type(dataObject) == datetime.datetime :
formField = widgets.NickDatetimeField(field_name=field.name, is_required=False, showsTime=True)
elif type(dataObject) == datetime.time :
formField = widgets.NickDatetimeField(field_name=field.name, is_required=False, showsTime=True, inputDate=False)
elif type(dataObject) == bool :
formField = forms.CheckboxField(field_name=field.name)
else :
if widgetType == wikidbase.core.context.MS_LIST :
formField = forms.SelectMultipleField(field_name=field.name, is_required=False, choices=choices)
elif widgetType == wikidbase.core.context.SS_LIST :
formField = widgets.EditableSelectField(field_name=field.name, is_required=False, choices=choices)
elif field.nName in ["password"] :
formField = forms.PasswordField(field_name=field.name, is_required=False)
else :
formField = widgets.TextFieldExtra(field_name=field.name, is_required=False)
return formField
def save(self, newData) :
debugOutput("Saving data: '%s'" % newData)
# raise Exception("newData %s" % newData.getlist("Home address"))
# Try to get the context.
wikidpageContext = None
if self.wikidpage.context :
wikidpageContexts = wikidbase.core.context.getContexts()
nContext = wikidbase.core.normaliseTerm(self.wikidpage.context)
if nContext in wikidpageContexts :
wikidpageContext = wikidpageContexts[nContext]
wikidpageFields = self.wikidpage.getFields().getDict()
for nFieldName in wikidpageFields :
field = wikidpageFields[nFieldName]
if not wikidbase.core.pagecontent.isDataField(field):
continue
# Get the data object from either the data or the context.
if field.data :
dataObject = datatype.convert(field.data, mode=datatype.TO_PYTHON)
else :
try :
dataObject = datatype.convert(wikidpageContext.dataFieldContexts[field.nName].representativeData.stack[0], mode=datatype.TO_PYTHON)
except :
dataObject = None
debugOutput("data object for '%s' is type '%s'" % (field.name, type(dataObject)))
# TODO: Is this field an ms list.
nContext = wikidbase.core.context.normaliseTerm(self.wikidpage.context)
nFieldName = field.nName
contextFieldState = wikidbase.core.state.getContextFieldState(nContext, nFieldName)
try :
widgetType = contextFieldState[wikidbase.core.context.LIST_STYLE]
except :
widgetType = wikidbase.core.context.AUTO_WIDGET
if field.name in newData :
data = newData.getlist(field.name)
debugOutput("fname %s type %s data %s" % (field.name, type(data), data))
if type(data) == list :
data = "\n".join(data)
debugOutput("joined data %s" % data)
# Double convert the data.
field.data = datatype.convert(datatype.convert(data, mode=datatype.TO_PYTHON), mode=datatype.TO_STRING)
# Interperet non-posted boolean data.
elif type(dataObject) == bool :
field.data = datatype.convert(False, mode=datatype.TO_STRING)
elif widgetType == wikidbase.core.context.MS_LIST :
field.data = ""
# Interpret non-posted multi-select as an empty set.
# TODO: What happens when no list is posted for ms select.
self.wikidpage.save()
# Ensure the wikidpage is given a name.
if not self.wikidpage.name :
self.wikidpage.name = "%s-%s" % (wikidbase.core.normaliseTerm(self.wikidpage.context) or DEFAULT_PAGE_NAME,self.wikidpage.id)
self.wikidpage.save()
return self.wikidpage
def isValidTime(field_data, all_data):
if not type(datatype.convert(field_data,mode=datatype.TO_PYTHON)) == datetime.time:
raise validators.ValidationError, 'Enter a valid time.'
def isValidDatetime(field_data, all_data):
debugOutput("isValidTime %s" % field_data)
if not type(datatype.convert(field_data,mode=datatype.TO_PYTHON)) == datetime.datetime:
raise validators.ValidationError, 'Enter a valid date <b>and</b> time.'
def isValidTime(field_data, all_data):
if not type(datatype.convert(field_data,
mode=datatype.TO_PYTHON)) == datetime.time:
raise validators.ValidationError, 'Enter a valid time.'
def isValidDatetime(field_data, all_data):
debugOutput("isValidTime %s" % field_data)
if not type(datatype.convert(
field_data, mode=datatype.TO_PYTHON)) == datetime.datetime:
raise validators.ValidationError, 'Enter a valid date <b>and</b> time.'