class NicoLoad_Load():
def __init__(self, log, status, databaseObj):
"""
log: modules.nicoLoad_base.Log()
"""
self.log = log
self.status = status
self.db = databaseObj
if not(self.db.extMovieTable() and self.db.extMylistTable() and self.db.extMylistHasMovieTable()):
self.db.ctTable()
self.queue = Queue()
self.__loading = False
self.waitTime = 1
# まとめ系
def nicoLoad_Load(self, *args):
"""
load全般まとめ関数
General Summary of load function
コマンドから実行する関数を振り分ける
Distributes the function to be executed from the command
"""
option = args[1]
if option.has_key(args[0]):
url = option[args[0]]
self.movieLoad(url)
else:
self.movieLoadAll()
def rss(self, *args):
"""
rss
-id mylistid_1, mylistid_2, ...
"""
option = args[1]
del option[args[0]]
if len(args) == 0:
rssAll()
else:
for mylistid in option["-id"]:
rssOne(mylistid)
def regNicoInfo(self, *args):
"""
ニコニコID,PW保存
save niconico ID & PW
"""
nicoid = raw_input(u"ニコニコ動画のIDを入力してください\n>>> ")
nicopw = raw_input(u"ニコニコ動画のパスワードを入力してください\n>>> ")
self.db.svNicoid(nicoid)
self.db.svNicopw(nicopw)
return True
def regGmailInfo(self, *args):
"""
GmailID,PWを保存
save Gmail ID & PW
"""
gmailid = raw_input(u"GmailのIDを入力してください\n>>> ")
gmailpw = raw_input(u"GmailのPWを入力してください\n>>> ")
self.db.svGmailid(gmailid)
self.db.svGmailpw(gmailpw)
def regSavedir(self, *args):
"""
保存フォルダを保存
save the folder for saving
"""
flag = False
while True:
savedir = raw_input(u"保存フォルダを入力してください\n>>> ")
flag = os.path.exists(os.path.abspath(savedir))
if flag:
break
else:
print u"入力されたフォルダは存在しません"
self.db.svSavedir(savedir)
return True
def getNicoInfo(self):
"""return (nicoid, nicopw)"""
return (self.db.getNicoid(), self.db.getNicopw())
def getGmailInfo(self):
"""return (gmailid, gmailpw)"""
return (self.db.getGmailid(), self.db.getGmailpw())
def getSavedir(self):
"""return savedir"""
return self.db.getSavedir()
def movieQueueAdd(self, movieid):
"""
保存キューに追加
add to the queue for string
"""
print "addQueue: %s" % movieid
self.queue.add(movieid)
def movieQueueAddFront(self, movieid):
"""
保存キューの先頭に追加
added to the top of the queue for string
"""
print "addQueue: %s" % movieid
self.queue.addFromnt(movieid)
def movieQueueaddNext(self, movieid):
"""
保存キューの次に追加
added to the queue for string the second
"""
print "addQueue: %s" % movieid
self.queue.addNext(movieid)
def movieQueueGet(self):
"""
キューから取得
retrived from the queue
"""
return self.queue.get()
def movieLoadDone(self):
"""
タスクの終了を伝える
convey the completetion of the task
タスクを実行したら呼び出す
call to perform the task
"""
self.queue.done()
self.status.chLoadingMovie()
def movieLoadLoop(self):
"""
保存用ループ
loop for strage
"""
# 既に動いていたら I was already running
if self.__loading: return False
else: self.__loading = True
while True:
try:
movieid = self.movieQueueGet()
self.status.setLoadingMovieid(movieid)
self.movieLoad()
except QueueEmptyError():
self.status.setLoadingMovieid(u"waiting")
time.sleep(self.waitTime)
def movieLoadAll(self):
"""
すべての動画を保存
save all videos
"""
for movieid in self.db.getMovieState0():
self.movieQueueAdd(movieid)
self.movieLoadLoop()
def movieLoad(self, movieid):
"""
動画を保存
save videos
"""
if not self.db.addMovie(movieid): # DB登録
# 登録済みの時
self.db.setMovieState(movieid, 0)
api = NicovideoAPI(movie_id=movieid)
title = api.get_movie_title()
# ext = api.get_movie_type()
nicoid = self.db.getNicoid()
nicopw = self.db.getNicopw()
savedir = getSavedir()
self.status.chLoadingMovie()
self.db.setMovieTitle(movieid, title)
self.db.setMovieDescription(movieid, api.get_movie_description())
self.db.setMovieLength(movieid, api.get_movie_length())
self.db.setMoviePath(movieid, moviePath)
self.status.setLoadingMovieid(movieid)
moviePath = api.save_movie(nicoid, nicopw, savedir)
self.db.setMovieThumbnail(movieid, api.save_thumbnail(savedir))
self.db.setMovieComment(movieid, api.save_comment(nicoid, nicopw, savedir, title))
self.db.setMovieState(movieid, 1)
self.movieLoadDone()
return True
def rssOne(self, mylistid):
"""
マイリスRSS
RSS for mylist
"""
various = Various(self.database)
for movieid in various.rsscheck(mylistid):
self.movieQueueAdd(movieid)
def rssAll(self):
"""
マイリスRSS
RSS for mylist
"""
various = Various(self.databse)
for mylistid in self.DB.getMylistAll():
for movieid in various.rsscheck(mylistid):
self.movieQueueAdd(movieid)
def exit(self):
"""
Exit
"""
self.db.close()
