bin/maintenance/category_integrity.py

   1 # This file is part of Indico.
   2 # Copyright (C) 2002 - 2015 European Organization for Nuclear Research (CERN).
   3 #
   4 # Indico is free software; you can redistribute it and/or
   5 # modify it under the terms of the GNU General Public License as
   6 # published by the Free Software Foundation; either version 3 of the
   7 # License, or (at your option) any later version.
   8 #
   9 # Indico is distributed in the hope that it will be useful, but
  10 # WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12 # General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU General Public License
  15 # along with Indico; if not, see <http://www.gnu.org/licenses/>.
  16
  17 """
  18 This simple script checks whether the `_numConferences` attribute is consistent for
  19 every category, and fixs them otherwise.
  20
  21 It can be added as a periodic cronjob.
  22 """
  23
  24 import sys
  25
  26 from indico.core.db import DBMgr
  27 from MaKaC.conference import CategoryManager, ConferenceHolder
  28
  29
  30
  31 def set_num_conferences(categ, num):
  32     categ._numConferences = num
  33
  34
  35 def check_event_number_consistency(root):
  36
  37     for categ in root.subcategories.itervalues():
  38         check_event_number_consistency(categ)
  39
  40     if root.conferences:
  41         return
  42     else:
  43         total = 0
  44         for scat in root.subcategories.itervalues():
  45             total += scat._numConferences
  46         if root._numConferences != total:
  47             sys.stderr.write("(n) '%s': expected %d, got %d! " % (root.getId(), total, root._numConferences))
  48             set_num_conferences(root, total)
  49             sys.stderr.write("[FIXED]\n")
  50             dbi.commit()
  51
  52
  53 def check_event_number_leaves(categ, expected, dbi):
  54     obtained = categ._numConferences
  55
  56     if expected != obtained:
  57         # something is wrong!
  58         sys.stderr.write( "(l) '%s': expected %d, got %d! " % (cid, expected, obtained))
  59
  60         # fix it
  61         set_num_conferences(categ, expected)
  62         sys.stderr.write("[FIXED]\n")
  63         dbi.commit()
  64
  65
  66 if __name__ == '__main__':
  67     dbi = DBMgr.getInstance()
  68     dbi.startRequest()
  69
  70     confIndex = ConferenceHolder()._getIdx()
  71     index = CategoryManager()._getIdx()
  72
  73     for cid, categ in index.iteritems():
  74         # conferece-containing categories
  75         if categ.conferences:
  76
  77             # since we're here check consistency of TreeSet
  78             categ.conferences._check()
  79
  80             lenConfs = 0
  81             for conf in categ.conferences:
  82                 lenConfs += 1
  83                 if conf.getId() not in confIndex:
  84                     sys.stderr.write("[%s] '%s' not in ConferenceHolder!\n" % (cid, conf.getId()))
  85
  86             # check event numbers
  87             check_event_number_leaves(categ, lenConfs, dbi)
  88
  89             # check that there are no subcategories
  90             if categ.subcategories:
  91                 sys.stderr.write("'%s' has subcategories AND conferences!\n")
  92
  93         # as for category-containing categories
  94         else:
  95             for subcat in categ.subcategories:
  96                 if subcat not in index:
  97                     sys.stderr.write("'%s' is a child of '%s' but is not in the CategoryManager" % (subcat, cid))
  98
  99         # check that the the category has a parent
 100         if categ.getOwner() == None and cid != '0':
 101             sys.stderr.write("'%s' has no owner!\n")
 102
 103         # check that the contained id is the same as the indexation key
 104         if categ.getId() != cid:
 105             sys.stderr.write("'%s' indexed as '%s'\n" % (categ.getId(), cid))
 106
 107     # now check that the event number totals are OK
 108     check_event_number_consistency(CategoryManager().getById('0'))