def __init__(self, criteria=1): self.db = DB() if criteria is None: criteria = 1 if isinstance(criteria, int): sqldata = self.db.get_downloaded_days(criteria) elif isinstance(criteria, datetime.datetime): sqldata = self.db.get_downloaded_date(criteria) elif isinstance(criteria, str): sqldata = self.db.get_downloaded_title(criteria) self.sqldata = sqldata
def __init__(self, dbdata=[], show_type='current'): typelist = ('new', 'nondb', 'current') if show_type not in typelist: raise Exception('incorrect show type') if show_type == 'current': self._set_db_data(dbdata) self._get_thetvdb_series_data() self.search_provider = Search() elif show_type == 'nondb': self.search_provider = Search() self.console_columns = Config.console_columns self.db = DB()