summaryrefslogtreecommitdiff
path: root/anidb
diff options
context:
space:
mode:
authorJon Bergli Heier <snakebite@jvnv.net>2010-10-21 19:15:11 +0200
committerJon Bergli Heier <snakebite@jvnv.net>2010-10-21 19:15:11 +0200
commit58da4e434d844da76226fdaceb6952532d68fb1e (patch)
tree29787310fd7b8ee591f953a45d9b23a6f4c00b83 /anidb
parent29a2c4c430cab5821580467ea69f8f9347d699b5 (diff)
Make code compatible with Python 3.x; should work with 2.6 and newer.
Diffstat (limited to 'anidb')
-rwxr-xr-xanidb54
1 files changed, 31 insertions, 23 deletions
diff --git a/anidb b/anidb
index b88e6cc..ffc27ea 100755
--- a/anidb
+++ b/anidb
@@ -1,9 +1,17 @@
#!/usr/bin/env python
import pyanidb, pyanidb.hash
-import ConfigParser, optparse, os, sys, getpass
+try:
+ import ConfigParser
+except ImportError:
+ import configparser as ConfigParser
+import optparse, os, sys, getpass
from collections import deque
+# Workaround for input/raw_input
+if hasattr(__builtins__, 'raw_input'):
+ input = raw_input
+
# Colors.
red = lambda x: '\x1b[1;31m' + x + '\x1b[0m'
@@ -60,7 +68,7 @@ if options.cache:
try:
import xattr
except ImportError:
- print red('No xattr, caching disabled.')
+ print(red('No xattr, caching disabled.'))
options.cache = False
options.identify = options.identify or options.rename
options.login = options.add or options.watched or options.identify
@@ -71,7 +79,7 @@ if not options.format:
if options.login:
if not options.username:
- options.username = raw_input('Username: ')
+ options.username = input('Username: ')
if not options.password:
options.password = getpass.getpass()
@@ -82,12 +90,12 @@ remaining = deque(args)
while remaining:
name = remaining.popleft()
if not os.access(name, os.R_OK):
- print red('Invalid file:'), name
+ print('{0} {1}'.format(red('Invalid file:'), name))
elif os.path.isfile(name):
files.append(name)
elif os.path.isdir(name):
if not options.recursive:
- print red('Is a directory:'), name
+ print('{0} {1}'.format(red('Is a directory:'), name))
else:
for sub in sorted(os.listdir(name)):
if sub.startswith('.'):
@@ -99,7 +107,7 @@ while remaining:
remaining.appendleft(sub)
if not files:
- print blue('Nothing to do.')
+ print(blue('Nothing to do.'))
sys.exit(0)
# Authorization.
@@ -108,15 +116,15 @@ if options.login:
a = pyanidb.AniDB(options.username, options.password)
try:
a.auth()
- print blue('Logged in as user:'), options.username
+ print('{0} {1}'.format(blue('Logged in as user:'), options.username))
except pyanidb.AniDBUserError:
- print red('Invalid username/password.')
+ print(red('Invalid username/password.'))
sys.exit(1)
except pyanidb.AniDBTimeout:
- print red('Connection timed out.')
+ print(red('Connection timed out.'))
sys.exit(1)
- except pyanidb.AniDBError, e:
- print red('Fatal error:'), e
+ except pyanidb.AniDBError as e:
+ print('{0} {1}'.format(red('Fatal error:'), e))
sys.exit(1)
# Hashing.
@@ -124,7 +132,7 @@ if options.login:
hashed = unknown = 0
for file in pyanidb.hash.hash_files(files, options.cache, (('ed2k', 'md5', 'sha1', 'crc32') if options.multihash else ('ed2k',))):
- print blue('Hashed:'), 'ed2k://|file|%s|%d|%s|%s' % (file.name, file.size, file.ed2k, ' (cached)' if file.cached else '')
+ print('{0} ed2k://|file|{1}|{2}|{3}|{4}'.format(blue('Hashed:'), file.name, file.size, file.ed2k, ' (cached)' if file.cached else ''))
fid = (file.size, file.ed2k)
hashed += 1
@@ -132,16 +140,16 @@ for file in pyanidb.hash.hash_files(files, options.cache, (('ed2k', 'md5', 'sha1
# Multihash.
if options.multihash:
- print blue('MD5:'), file.md5
- print blue('SHA1:'), file.sha1
- print blue('CRC32:'), file.crc32
+ print('{0} {1}'.format(blue('MD5:'), file.md5))
+ print('{0} {1}'.format(blue('SHA1:'), file.sha1))
+ print('{0} {1}'.format(blue('CRC32:'), file.crc32))
# Identify.
if options.identify:
info = a.get_file(fid, ('gtag', 'romaji', 'epno', 'state', 'epromaji', 'crc32', 'filetype'), True)
fid = int(info['fid'])
- print green('Identified:'), '[%s] %s - %s - %s' % (info['gtag'], info['romaji'], info['epno'], info['epromaji'])
+ print('{0} [{1}] {2} - {3} - {4}'.format(green('Identified:'), info['gtag'], info['romaji'], info['epno'], info['epromaji']))
# Renaming.
@@ -155,34 +163,34 @@ for file in pyanidb.hash.hash_files(files, options.cache, (('ed2k', 'md5', 'sha1
'crc': info['crc32'],
'CRC': info['crc32'].upper(),
'suf': info['filetype']}
- for name, value in rename_data.iteritems():
+ for name, value in rename_data.items():
s = s.replace(r'%' + name, value)
if s[0] == '_':
s = s[1:].replace(' ', '_')
s = s.replace('/', '_')
- print yellow('Renaming to:'), s
+ print('{0} {1}'.format(yellow('Renaming to:'), s))
os.rename(file.name, os.path.join(os.path.split(file.name)[0], s))
# Adding.
if options.add:
a.add_file(fid, viewed = options.watched, retry = True)
- print green('Added to mylist.')
+ print(green('Added to mylist.'))
# Watched.
elif options.watched:
a.add_file(fid, viewed = True, edit = True, retry = True)
- print green('Marked watched.')
+ print(green('Marked watched.'))
except pyanidb.AniDBUnknownFile:
- print red('Unknown file.')
+ print(red('Unknown file.'))
unknown += 1
except pyanidb.AniDBNotInMylist:
- print red('File not in mylist.')
+ print(red('File not in mylist.'))
# Finished.
-print blue('Hashed %d files%s.' % (hashed, (', %d unknown' % unknown) if unknown else ''))
+print(blue('Hashed {0} files{1}.'.format(hashed, ', {0} unknown'.format(unknown) if unknown else '')))