forked from enlavin/tvscrap
-
Notifications
You must be signed in to change notification settings - Fork 0
/
tvscrap.py
executable file
·67 lines (55 loc) · 1.91 KB
/
tvscrap.py
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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
TVScrap.py
"""
# GNU General Public Licence (GPL)
#
# This program is free software; you can redistribute it and/or modify it under
# the terms of the GNU General Public License as published by the Free Software
# Foundation; either version 2 of the License, or (at your option) any later
# version.
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
# details.
# You should have received a copy of the GNU General Public License along with
# this program; if not, write to the Free Software Foundation, Inc., 59 Temple
# Place, Suite 330, Boston, MA 02111-1307 USA
import socket
import sys
import storm.locals as st
from db import connect_db
class TVScrap(object):
def __init__(self):
self.seriesdb = connect_db()
self.store = st.Store(self.seriesdb)
socket.setdefaulttimeout(60)
def get_command(self, name):
"""Find the module implementing a command and returns a initialized instance of Command"""
module = __import__('commands', {}, [], [name])
klass = getattr(getattr(module, name), 'Command')
return klass(self.store)
def run_command(self, cmdname=None):
try:
if cmdname is None:
cmdname = sys.argv[1]
cmd = self.get_command(cmdname)
except AttributeError:
self.run_command('help')
sys.exit(1)
if not cmd.check_args(sys.argv[1:]):
cmd.show_help()
sys.exit(1)
ret = cmd.run()
def run(self):
if len(sys.argv) == 1:
self.run_command('help')
sys.exit(1)
ret = self.run_command()
return ret
def main():
tvs = TVScrap()
tvs.run()
if __name__ == '__main__':
main()