/usr/share/pyshared/schooltool/person/sampledata.py is in python-schooltool 1:2.1.0-0ubuntu1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 | #
# SchoolTool - common information systems platform for school administration
# Copyright (c) 2007 Shuttleworth Foundation
#
# 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 2 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, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
#
"""
Person sample data generation
"""
import datetime
import os
from pytz import utc
from zope.interface import implements
from zope.security.proxy import removeSecurityProxy
from schooltool.person.person import Person
from schooltool.sampledata import PortableRandom
from schooltool.sampledata.interfaces import ISampleDataPlugin
from schooltool.sampledata.name import NameGenerator
from schooltool.group.interfaces import IGroupContainer
from schooltool.term.interfaces import ITermContainer
from schooltool.app.interfaces import ISchoolToolCalendar
from schooltool.app.cal import CalendarEvent
from schooltool.common import DateRange
class ChoiceGenerator(object):
def __init__(self, seed, choices):
self.random = PortableRandom(seed)
self.choices = choices
def generate(self):
return self.random.choice(self.choices)
class SampleStudents(object):
implements(ISampleDataPlugin)
name = 'students'
dependencies = ('terms', )
# Number of persons to generate
power = 1000
def personFactory(self, namegen, prefixgen, gendergen, count):
first_name, last_name, full_name = namegen.generate()
person_id = 'student%03d' % count
person = Person(person_id, full_name)
person.setPassword(person_id)
return person
def generate(self, app, seed=None):
namegen = NameGenerator(str(seed) + self.name)
prefixgen = ChoiceGenerator(str(seed), ['Mr', 'Mrs', 'Miss', ''])
gendergen = ChoiceGenerator(str(seed), ['male', 'female'])
students = IGroupContainer(app)['students']
for count in range(self.power):
person = self.personFactory(namegen, prefixgen, gendergen, count)
# Without removeSecurityProxy we can't add members a
# group.
removeSecurityProxy(students.members).add(person)
app['persons'][person.__name__] = person
class SampleTeachers(object):
implements(ISampleDataPlugin)
name = 'teachers'
dependencies = ('terms', )
# Number of teachers to generate
power = 48
def personFactory(self, namegen, count):
first_name, last_name, full_name = namegen.generate()
person_id = 'teacher%03d' % count
person = Person(person_id, full_name)
person.setPassword(person_id)
return person
def generate(self, app, seed=None):
namegen = NameGenerator(str(seed) + self.name)
teachers = IGroupContainer(app)['teachers']
for count in range(self.power):
person = self.personFactory(namegen, count)
# Without removeSecurityProxy we can't add members a
# group.
removeSecurityProxy(teachers.members).add(person)
app['persons'][person.__name__] = person
class SamplePersonalEvents(object):
"""Sample data plugin class that generates personal random events."""
implements(ISampleDataPlugin)
name = 'personal_events'
dependencies = ('students', 'teachers', 'terms')
probability = 2 # probability of having an event on any day
def _readLines(self, filename):
"""Read in lines from file
Filename is relative to the module.
Returned lines are stripped.
"""
fullpath = os.path.join(os.path.dirname(__file__), filename)
lines = file(fullpath).readlines()
return [line.strip() for line in lines]
def generate(self, app, seed=None):
self.random = PortableRandom(seed)
event_titles = self._readLines('event_titles.txt')
person_ids = [person for person in app['persons'].keys()
if person.startswith('student') or
person.startswith('teacher')]
dates = []
for term in ITermContainer(app).values():
dates.append(term.first)
dates.append(term.last)
first = min(dates)
last = max(dates)
days = DateRange(first, last)
for person_id in person_ids:
person = app['persons'][person_id]
calendar = ISchoolToolCalendar(person)
for day in days:
if self.random.randrange(0, 100) < self.probability:
event_title = self.random.choice(event_titles)
time_hour = self.random.randint(6, 23)
time_min = self.random.choice((0, 30))
event_time = datetime.datetime(day.year,
day.month,
day.day,
time_hour,
time_min,
tzinfo=utc)
event_duration = datetime.timedelta(
minutes=self.random.randint(1, 12)*30)
event = CalendarEvent(event_time,
event_duration,
event_title)
calendar.addEvent(event)
|