Refactor nzbToMediaDB -> main_db

TorrentToMedia.py

@@ -5,7 +5,7 @@ import os
 import sys
 
 import core
-from core import logger, nzbToMediaDB
+from core import logger, main_db
 from core.nzbToMediaUserScript import external_script
 from core.nzbToMediaUtil import char_replace, convert_to_ascii, plex_update, replace_links
 from six import text_type
@@ -19,7 +19,7 @@ def process_torrent(input_directory, input_name, input_category, input_hash, inp
     if client_agent != 'manual' and not core.DOWNLOADINFO:
         logger.debug('Adding TORRENT download info for directory {0} to database'.format(input_directory))
 
-        my_db = nzbToMediaDB.DBConnection()
+        my_db = main_db.DBConnection()
 
         input_directory1 = input_directory
         input_name1 = input_name

core/__init__.py

@@ -42,7 +42,7 @@ MYAPP = None
 import six
 from six.moves import reload_module
 
-from core import logger, nzbToMediaDB, versionCheck, databases, transcoder
+from core import logger, main_db, versionCheck, databases, transcoder
 from core.auto_process import Comic, Game, Movie, Music, TV
 from core.configuration import config
 from core.nzbToMediaUtil import (
@@ -319,7 +319,7 @@ def initialize(section=None):
             logger.info("{0}: {1}".format(item, os.environ[item]), "ENVIRONMENT")
 
     # initialize the main SB database
-    nzbToMediaDB.upgrade_database(nzbToMediaDB.DBConnection(), databases.InitialSchema)
+    main_db.upgrade_database(main_db.DBConnection(), databases.InitialSchema)
 
     # Set Version and GIT variables
     NZBTOMEDIA_VERSION = '11.06'

core/databases.py

@@ -1,6 +1,6 @@
 # coding=utf-8
 
-from core import logger, nzbToMediaDB
+from core import logger, main_db
 from core.nzbToMediaUtil import backup_versioned_file
 
 MIN_DB_VERSION = 1  # oldest db version we support migrating from
@@ -9,7 +9,7 @@ MAX_DB_VERSION = 2
 
 def backup_database(version):
     logger.info("Backing up database before upgrade")
-    if not backup_versioned_file(nzbToMediaDB.db_filename(), version):
+    if not backup_versioned_file(main_db.db_filename(), version):
         logger.log_error_and_exit("Database backup failed, abort upgrading database")
     else:
         logger.info("Proceeding with upgrade")
@@ -20,7 +20,7 @@ def backup_database(version):
 # ======================
 # Add new migrations at the bottom of the list; subclass the previous migration.
 
-class InitialSchema(nzbToMediaDB.SchemaUpgrade):
+class InitialSchema(main_db.SchemaUpgrade):
     def test(self):
         no_update = False
         if self.has_table("db_version"):

core/nzbToMediaUtil.py

@@ -26,7 +26,7 @@ from transmissionrpc.client import Client as TransmissionClient
 from utorrent.client import UTorrentClient
 
 import core
-from core import logger, nzbToMediaDB
+from core import logger, main_db
 from core.extractor import extractor
 
 requests.packages.urllib3.disable_warnings()
@@ -1284,7 +1284,7 @@ def backup_versioned_file(old_file, version):
 def update_download_info_status(input_name, status):
     logger.db("Updating status of our download {0} in the DB to {1}".format(input_name, status))
 
-    my_db = nzbToMediaDB.DBConnection()
+    my_db = main_db.DBConnection()
     my_db.action("UPDATE downloads SET status=?, last_update=? WHERE input_name=?",
                  [status, datetime.date.today().toordinal(), text_type(input_name)])
 
@@ -1292,7 +1292,7 @@ def update_download_info_status(input_name, status):
 def get_download_info(input_name, status):
     logger.db("Getting download info for {0} from the DB".format(input_name))
 
-    my_db = nzbToMediaDB.DBConnection()
+    my_db = main_db.DBConnection()
     sql_results = my_db.select("SELECT * FROM downloads WHERE input_name=? AND status=?",
                                [text_type(input_name), status])
 

nzbToMedia.py

@@ -628,7 +628,7 @@ import os
 import sys
 
 import core
-from core import logger, nzbToMediaDB
+from core import logger, main_db
 from core.auto_process import Comic, Game, Movie, Music, TV
 from core.nzbToMediaUserScript import external_script
 from core.nzbToMediaUtil import char_replace, clean_dir, convert_to_ascii, extract_files, get_dirs, get_download_info, get_nzoid, plex_update, update_download_info_status
@@ -653,7 +653,7 @@ def process(input_directory, input_name=None, status=0, client_agent='manual', d
     if client_agent != 'manual' and not core.DOWNLOADINFO:
         logger.debug('Adding NZB download info for directory {0} to database'.format(input_directory))
 
-        my_db = nzbToMediaDB.DBConnection()
+        my_db = main_db.DBConnection()
 
         input_directory1 = input_directory
         input_name1 = input_name