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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
#!/usr/bin/env python
import pyanidb, pyanidb.hash
import ConfigParser, optparse, os, sys, getpass, multihash
# Config.
config = {}
try:
cp = ConfigParser.ConfigParser()
cp.read(os.path.expanduser('~/.pyanidb.conf'))
for option in cp.options('pyanidb'):
config[option] = cp.get('pyanidb', option)
except:
pass
# Options.
op = optparse.OptionParser()
op.add_option('-u', '--username', help = 'AniDB username.',
action = 'store', dest = 'username', default = config.get('username'))
op.add_option('-p', '--password', help = 'AniDB password.',
action = 'store', dest = 'password', default = config.get('password'))
op.add_option('-r', '--recursive', help = 'Recurse into directories.',
action = 'store_true', dest = 'recursive', default = False)
op.add_option('-s', '--suffix', help = 'File suffix for recursive matching.',
action = 'append', dest = 'suffix', default = config.get('suffix', '').split())
op.add_option('-i', '--identify', help = 'Identify files.',
action = 'store_true', dest = 'identify', default = False)
op.add_option('-a', '--add', help = 'Add files to mylist.',
action = 'store_true', dest = 'add', default = False)
op.add_option('-n', '--rename', help = 'Rename files.',
action = 'store_true', dest = 'rename', default = False)
op.add_option('-f', '--format', help = 'Filename format.',
action = 'store', dest = 'format', default = config.get('format'))
options, args = op.parse_args(sys.argv[1:])
# Defaults.
options.identify = options.identify or options.rename
options.login = options.add or options.identify
if not options.suffix:
options.suffix = ['avi', 'ogm', 'mkv']
if not options.format:
options.format = r'_[%group]_%anime_-_%epno%ver_[%CRC].%suf'
if options.login:
if not options.username:
options.username = raw_input('Username: ')
if not options.password:
options.passord = getpass.getpass()
# Input files.
files = []
for name in args:
if not os.access(name, os.R_OK):
print 'Invalid file: %s' % (name)
elif os.path.isfile(name):
files.append(name)
elif os.path.isdir(name):
if not options.recursive:
print 'Is a directory: %s' % (name)
else:
for root, subdirs, subfiles in os.walk(name):
subdirs.sort()
subfiles.sort()
files += [os.path.join(root, file) for file in subfiles if sum([file.endswith('.' + suffix) for suffix in options.suffix])]
if not files:
print 'All operations finished.'
sys.exit(0)
# Authorization.
if options.login:
a = pyanidb.AniDB(options.username, options.password)
try:
a.auth()
print 'Logged in as user %s.' % (options.username)
except pyanidb.AniDBUserError:
print 'Invalid username/password.'
sys.exit(1)
except pyanidb.AniDBTimeout:
print 'Connection timed out.'
sys.exit(1)
except pyanidb.AniDBError, e:
print 'Fatal error:', e
sys.exit(1)
# Hashing.
for filename, hash in pyanidb.hash.hash_files(files):
size = os.path.getsize(filename)
print 'Hashed: ed2k://|file|%s|%d|%s|' % (filename, size, hash.ed2k())
fid = (size, hash.ed2k())
try:
# Identify.
if options.identify:
info = a.get_file(fid, ('gtag', 'romaji', 'epno', 'state', 'epromaji', 'crc32', 'filetype'), True)
fid = int(info['fid'])
print 'Identified: [%s] %s - %s - %s' % (info['gtag'], info['romaji'], info['epno'], info['epromaji'])
# Renaming.
if options.rename:
s = options.format
rename_data = {
'group': info['gtag'],
'anime': info['romaji'],
'epno': info['epno'],
'ver': {0: '', 4: 'v2', 8: 'v3', 16: 'v4', 32: 'v5'}[(int(info['state']) & 0x3c)],
'crc': info['crc32'],
'CRC': info['crc32'].upper(),
'suf': info['filetype']}
for name, value in rename_data.iteritems():
s = s.replace(r'%' + name, value)
if s[0] == '_':
s = s[1:].replace(' ', '_')
s = s.replace('/', '_')
print 'Renaming to: %s' % (s)
os.rename(filename, os.path.join(os.path.split(filename)[0], s))
# Adding.
if options.add:
a.add_file(fid, retry = True)
print 'Added to mylist.'
except pyanidb.AniDBUnknownFile:
print 'Unknown file.'
# Finished.
print 'All operations finished.'
|
