Reports: Fix InternalDatabaseVersion

You could not import a class, that is defined in a function. We need to
return the class, but we do not want to define it multiple times. So we
have to save the class in a global variable.

1 files changed, 4 insertions, 4 deletions

diff --git a/src/lib/Bcfg2/Server/Reports/updatefix.py b/src/lib/Bcfg2/Server/Reports/updatefix.py
index 91c370994..09b218464 100644
--- a/src/lib/Bcfg2/Server/Reports/updatefix.py
+++ b/src/lib/Bcfg2/Server/Reports/updatefix.py
@@ -4,7 +4,7 @@ import django.core.management
 import sys
 import logging
 import traceback
-from Bcfg2.Server.models import InternalDatabaseVersion
+from Bcfg2.Server.models import internal_database_version
 logger = logging.getLogger('Bcfg2.Server.Reports.UpdateFix')
 
 
@@ -138,7 +138,7 @@ def rollupdate(current_version):
                              exc_info=1)
             # since array start at 0 but version start at 1
             # we add 1 to the normal count
-            ret = InternalDatabaseVersion.objects.create(version=i + 1)
+            ret = internal_database_version().create(version=i + 1)
         return ret
     else:
         return None
@@ -149,10 +149,10 @@ def update_database():
     try:
         logger.debug("Running upgrade of models to the new one")
         django.core.management.call_command("syncdb", interactive=False, verbosity=0)
-        know_version = InternalDatabaseVersion.objects.order_by('-version')
+        know_version = internal_database_version().order_by('-version')
         if not know_version:
             logger.debug("No version, creating initial version")
-            know_version = InternalDatabaseVersion.objects.create(version=lastversion)
+            know_version = internal_database_version().create(version=lastversion)
         else:
             know_version = know_version[0]
         logger.debug("Presently at %s" % know_version)