database manage init

couchpotato/core/plugins/release/index.py

@@ -4,6 +4,7 @@ from CodernityDB.tree_index import TreeBasedIndex
 
 
 class ReleaseIndex(TreeBasedIndex):
+    _version = 1
 
     def __init__(self, *args, **kwargs):
         kwargs['key_format'] = '32s'
@@ -30,6 +31,7 @@ class ReleaseIndex(TreeBasedIndex):
 
 
 class ReleaseStatusIndex(TreeBasedIndex):
+    _version = 1
 
     def __init__(self, *args, **kwargs):
         kwargs['key_format'] = '32s'
@@ -44,6 +46,7 @@ class ReleaseStatusIndex(TreeBasedIndex):
 
 
 class ReleaseIDIndex(HashIndex):
+    _version = 1
 
     def __init__(self, *args, **kwargs):
         kwargs['key_format'] = '32s'
@@ -58,6 +61,7 @@ class ReleaseIDIndex(HashIndex):
 
 
 class ReleaseDownloadIndex(HashIndex):
+    _version = 1
 
     def __init__(self, *args, **kwargs):
         kwargs['key_format'] = '32s'

couchpotato/core/plugins/release/main.py

@@ -17,6 +17,13 @@ log = CPLog(__name__)
 
 class Release(Plugin):
 
+    _database = {
+        'release': ReleaseIndex,
+        'release_status': ReleaseStatusIndex,
+        'release_identifier': ReleaseIDIndex,
+        'release_download': ReleaseDownloadIndex
+    }
+
     def __init__(self):
         addApiView('release.manual_download', self.manualDownload, docs = {
             'desc': 'Send a release manually to the downloaders',
@@ -45,44 +52,10 @@ class Release(Plugin):
         addEvent('release.clean', self.clean)
         addEvent('release.update_status', self.updateStatus)
 
-        addEvent('database.setup', self.databaseSetup)
-
         # Clean releases that didn't have activity in the last week
         addEvent('app.load', self.cleanDone)
         fireEvent('schedule.interval', 'movie.clean_releases', self.cleanDone, hours = 4)
 
-    def databaseSetup(self):
-
-        db = get_db()
-
-        # Release media_id index
-        try:
-            db.add_index(ReleaseIndex(db.path, 'release'))
-        except:
-            log.debug('Index already exists')
-            db.edit_index(ReleaseIndex(db.path, 'release'))
-
-        # Release status index
-        try:
-            db.add_index(ReleaseStatusIndex(db.path, 'release_status'))
-        except:
-            log.debug('Index already exists')
-            db.edit_index(ReleaseStatusIndex(db.path, 'release_status'))
-
-        # Release identifier index
-        try:
-            db.add_index(ReleaseIDIndex(db.path, 'release_identifier'))
-        except:
-            log.debug('Index already exists')
-            db.edit_index(ReleaseIDIndex(db.path, 'release_identifier'))
-
-        # Release identifier index
-        try:
-            db.add_index(ReleaseDownloadIndex(db.path, 'release_download'))
-        except:
-            log.debug('Index already exists')
-            db.edit_index(ReleaseDownloadIndex(db.path, 'release_download'))
-
     def cleanDone(self):
         log.debug('Removing releases from dashboard')