dinislage/CouchPotatoServer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Migrate library file info

Browse Source
This commit is contained in:
Ruud
2014-03-05 15:41:16 +01:00
parent b25e0ea393
commit 81ba95f540
1 changed files with 45 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
couchpotato/core/database.py
+45 -3
View File
@@ -111,6 +111,7 @@ class Database(object):
return results
def migrate(self):
time.sleep(1)
from couchpotato import Env
@@ -131,10 +132,13 @@ class Database(object):
'movie': ['id', 'last_edit', 'library_id', 'status_id', 'profile_id', 'category_id'],
'library': ['id', 'identifier', 'info'],
'librarytitle': ['id', 'title', 'default', 'libraries_id'],
'library_files__file_library': ['library_id', 'file_id'],
'release': ['id', 'identifier', 'movie_id', 'status_id', 'quality_id', 'last_edit'],
'releaseinfo': ['id', 'identifier', 'value', 'release_id'],
'status': ['id', 'identifier'],
'properties': ['id', 'identifier', 'value'],
'file': ['id', 'path', 'type_id'],
'filetype': ['identifier', 'id']
}
migrate_data = {}
@@ -147,7 +151,13 @@ class Database(object):
columns = {}
for row in migrate_list[ml]:
columns[row] = p[rows.index(row)]
migrate_data[ml][p[0]] = columns
if not migrate_data[ml].get(p[0]):
migrate_data[ml][p[0]] = columns
else:
if not isinstance(migrate_data[ml][p[0]], list):
migrate_data[ml][p[0]] = [migrate_data[ml][p[0]]]
migrate_data[ml][p[0]].append(columns)
db = self.getDB()
@@ -268,12 +278,17 @@ class Database(object):
# Media
statuses = migrate_data['status']
libraries = migrate_data['library']
library_files = migrate_data['library_files__file_library']
all_files = migrate_data['file']
poster_type = migrate_data['filetype']['poster']
medias = migrate_data['movie']
media_link = {}
for x in medias:
m = medias[x]
status = statuses.get(m['status_id']).get('identifier')
# Only migrate wanted movies
if status != 'active': continue
l = libraries[m['library_id']]
@@ -281,6 +296,9 @@ class Database(object):
category_id = category_link.get(m['category_id'])
title = titles_by_library.get(m['library_id'])
releases = releases_by_media.get(x, [])
files = library_files.get(m['library_id'], [])
if not isinstance(files, list):
files = [files]
added_media = fireEvent('movie.add', {
'info': json.loads(l.get('info', '')),
@@ -290,6 +308,17 @@ class Database(object):
'title': title
}, force_readd = False, search_after = False, status = status, single = True)
for f in files:
file = all_files[f.get('file_id')]
# Only migrate posters
if file.get('type_id') == poster_type.get('id'):
if not added_media['files'].get('image_poster'):
added_media['files']['image_poster'] = []
if file.get('path') not in added_media['files']['image_poster']:
added_media['files']['image_poster'].append(file.get('path'))
db.update(added_media)
for rel in releases:
if not rel.get('info'): continue
@@ -297,5 +326,18 @@ class Database(object):
added_rel = fireEvent('release.create_from_search', [rel['info']], added_media, quality, single = True)
# Update status
added_id = db.get('release_identifier', added_rel[0])
fireEvent('release.update_status', added_id.get('_id'), statuses[rel.get('status_id')].get('identifier'))
added_rel['status'] = status
added_rel['last_edit'] = int(time.time())
db.update(added_rel)
break
return
# rename old database
os.rename(old_db, old_db + '.old')
if os.path.isfile(old_db + '-wal'):
os.rename(old_db + '-wal', old_db + '-wal.old')
if os.path.isfile(old_db + '-shm'):
os.rename(old_db + '-shm', old_db + '-shm.old')