py_tasks_melange: comparison scripts/stats.py

equal deleted inserted replaced

-:6c57c482f913
+:040359ccd20b
 def wrapped():
 gen = lambda: model.all()
 it = interactive.deepFetch(gen)
-entities = [(i.key().name(), i) for i in it]
+entities = [(i.key().id_or_name(), i) for i in it]
 return dict(entities)
 return wrapped
 print "deleted %d entities" % count
 print "Done"
 def loadPickle(name):
 """Loads a pickle.
 """
 f = open(name + '.dat')
 f = open("%s.dat" % name, 'w')
 cPickle.dump(target, f)
+def addOrganizationToSurveyRecords(survey_record_model):
+"""Set Organization in SurveyRecords entities of a given model.
+"""
+print "Fetching %s." % survey_record_model.__name__
+getSurveyRecord = getEntities(survey_record_model)
+survey_records = getSurveyRecord()
+survey_records_amount = len(survey_records)
+print "Fetched %d %s." % (survey_records_amount, survey_record_model.__name__)
+counter = 0
+for key in survey_records.keys():
+survey_records[key].org = survey_records[key].project.scope
+survey_records[key].put()
+counter += 1
+print str(counter) + '/' + str(survey_records_amount) + ' ' + str(key)
+print "Organization added to all %s." % survey_record_model.__name__
+def setOrganizationInSurveyRecords():
+"""Sets Organization property in ProjectSurveyRecords
+and GradingProjectSurveyRecords entities.
+"""
+from soc.models.project_survey_record import ProjectSurveyRecord
+from soc.models.grading_project_survey_record \
+import GradingProjectSurveyRecord
+addOrganizationToSurveyRecords(ProjectSurveyRecord)
+addOrganizationToSurveyRecords(GradingProjectSurveyRecord)
 def acceptedStudentsCSVExport(csv_filename, program_key_name):
 """Exports all accepted Students for particular program into CSV file.
 """
 # TODO(Pawel.Solyga): Add additional Program parameter to this method
 # so we export students from different programs
-# TODO(Pawel.SOlyga): Make it universal so it works with both GHOP
+# TODO(Pawel.Solyga): Make it universal so it works with both GHOP
 # and GSoC programs
 from soc.models.student_project import StudentProject
 from soc.models.student import Student
 from soc.models.organization import Organization
 'getMentors': getEntities(Mentor),
 'getOrgAdmins': getEntities(OrgAdmin),
 'getStudentProjects': getEntities(StudentProject),
 'getProps': getProps,
 'countStudentsWithProposals': countStudentsWithProposals,
+'setOrganizationInSurveyRecords': setOrganizationInSurveyRecords,
 'convertProposals': convertProposals,
 'addFollower': addFollower,
 'Organization': Organization,
 'Job': Job,
 'User': User,

changeset 2592	040359ccd20b
parent 2345	f78caf12f32d
child 2642	1bb33f56a3f4