1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
/****************************************************************************
**
** Copyright (C) 2015 Pelagicore AG
** Contact: http://www.qt.io/ or http://www.pelagicore.com/
**
** This file is part of the Neptune IVI UI.
**
** $QT_BEGIN_LICENSE:GPL3-PELAGICORE$
** Commercial License Usage
** Licensees holding valid commercial Pelagicore Neptune IVI UI
** licenses may use this file in accordance with the commercial license
** agreement provided with the Software or, alternatively, in accordance
** with the terms contained in a written agreement between you and
** Pelagicore. For licensing terms and conditions, contact us at:
** http://www.pelagicore.com.
**
** GNU General Public License Usage
** Alternatively, this file may be used under the terms of the GNU
** General Public License version 3 as published by the Free Software
** Foundation and appearing in the file LICENSE.GPLv3 included in the
** packaging of this file. Please review the following information to
** ensure the GNU General Public License version 3 requirements will be
** met: http://www.gnu.org/licenses/gpl-3.0.html.
**
** $QT_END_LICENSE$
**
** SPDX-License-Identifier: GPL-3.0
**
****************************************************************************/
"""
# scanmovies.py
# scans the ~/media/movies folder and generates all covers
# and a ~/media.db database with alll tracks.
import os
import sqlite3 as db
import argparse
from mutagen.mp4 import MP4
parser = argparse.ArgumentParser(description='scan movie folder')
parser.add_argument('--source', default='~/media/movies', help='movie dir to scan')
args = parser.parse_args()
source = os.path.expanduser(args.source)
albums = set()
connection = db.connect(os.path.expanduser('~/media.db'))
cursor = connection.cursor()
sql_drop = 'drop table if exists movies'
sql_create = """
create table movies (
pk integer primary key,
title text,
year integer,
genre text,
desc text,
source text,
cover text
)
"""
sql_insert = """
insert into movies (title,year,genre,desc,source,cover) VALUES (?,?,?,?,?,?)
"""
def createTable():
print('create table')
cursor.execute(sql_drop)
cursor.execute(sql_create)
def extractTag(media, tag):
obj = media.get(tag)
if obj:
return ''.join(obj)#.encode('utf-8')
return ''
def scanFolder(source):
print('scan folder: ' + source)
start = os.path.abspath(source)
for root, dirnames, filenames in os.walk(start):
for filename in filenames:
if not filename[-3:] == 'm4v':
continue
filepath = os.path.join(root, filename)
folderpath = os.path.relpath(root, start)
name = filename[0:-4]
print('analyze: ' + filepath)
try:
movie = MP4(filepath)
except:
print 'error reading: ', filepath
continue
title = extractTag(movie, '\xa9nam')
year = extractTag(movie, '\xa9day')
genre = extractTag(movie, '\xa9gen')
coverData = movie['covr'][0]
cover = os.path.join(folderpath, 'cover.jpg')
desc = extractTag(movie, 'desc')
source = os.path.join(folderpath, filename)
coverArt = file(os.path.join(start, cover), 'w')
coverArt.write(coverData)
coverArt.close()
cursor.execute(sql_insert, (title, year, genre, desc, source, cover))
createTable()
scanFolder(source)
connection.commit()
|
