You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
1.9 KiB
66 lines
1.9 KiB
14 years ago
|
from couchpotato import get_session
|
||
|
|
from couchpotato.api import addApiView
|
||
|
|
from couchpotato.core.event import fireEvent, addEvent, fireEventAsync
|
||
|
|
from couchpotato.core.helpers.request import jsonified, getParams
|
||
|
|
from couchpotato.core.logger import CPLog
|
||
|
|
from couchpotato.core.plugins.base import Plugin
|
||
|
|
from couchpotato.core.settings.model import File
|
||
|
|
from couchpotato.environment import Env
|
||
|
|
import time
|
||
|
|
|
||
|
|
|
||
|
|
log = CPLog(__name__)
|
||
|
|
|
||
|
|
class Manage(Plugin):
|
||
|
|
|
||
|
|
last_update = 0
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
|
||
|
|
fireEvent('scheduler.interval', identifier = 'manage.update_library', handle = self.updateLibrary, hours = 2)
|
||
|
|
|
||
|
|
addEvent('manage.update', self.updateLibrary)
|
||
|
|
addApiView('manage.update', self.updateLibraryView)
|
||
|
|
|
||
|
|
if not Env.setting('development'):
|
||
|
|
addEvent('app.load', self.updateLibrary)
|
||
|
|
|
||
|
|
def updateLibraryView(self):
|
||
|
|
|
||
|
|
params = getParams()
|
||
|
|
|
||
|
|
fireEventAsync('manage.update', full = params.get('full', True))
|
||
|
|
|
||
|
|
return jsonified({
|
||
|
|
'success': True
|
||
|
|
})
|
||
|
|
|
||
|
|
|
||
|
|
def updateLibrary(self, full = False):
|
||
|
|
|
||
|
|
if self.isDisabled() or (self.last_update > time.time() - 20):
|
||
|
|
return
|
||
|
|
|
||
|
|
directories = self.directories()
|
||
|
|
|
||
|
|
log.info('Updating manage library: %s' % directories)
|
||
|
|
|
||
|
|
for directory in directories:
|
||
|
|
fireEvent('scanner.folder', folder = directory)
|
||
|
|
|
||
|
|
# If cleanup option is enabled, remove offline files from database
|
||
|
|
if self.conf('cleanup'):
|
||
|
|
db = get_session()
|
||
|
|
files_in_path = db.query(File).filter(File.path.like(directory + '%%')).filter_by(available = 0).all()
|
||
|
|
[db.delete(x) for x in files_in_path]
|
||
|
|
db.commit()
|
||
|
|
db.remove()
|
||
|
|
|
||
|
|
self.last_update = time.time()
|
||
|
|
|
||
|
|
def directories(self):
|
||
|
|
try:
|
||
|
|
return self.conf('library', default = '').split('::')
|
||
|
|
except:
|
||
|
|
return []
|