1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283 |
- #
- # This file is part of stov, written by Helmut Pozimski 2012-2015.
- #
- # stov 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, version 2 of the License.
- #
- # stov 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 stov. If not, see <http://www.gnu.org/licenses/>.
- # -*- coding: utf8 -*-
- """This module takes care of managing and downloading single videos."""
- import os
- import sys
- import subprocess
- from lib_stov import stov_exceptions
- class Video(object):
- """This class stores all the attributes of a single youtube video and is
- also able to download it using youtube-dl.
- """
- def __init__(self, title, description, ytid, conf, downloaded, failcount=0,
- video_id=0):
- self._id = video_id
- self.title = title
- self.description = description
- self.ytid = ytid
- self.__conf = conf
- self.downloaded = downloaded
- self.failcnt = int(failcount)
- def download_video(self, directory, itag_value, video_codec):
- """Downloads the video by calling youtube-dl as an external process"""
- targetdir = self.__conf.values["downloaddir"] + "/" + directory
- if os.access(targetdir, os.F_OK) is False:
- try:
- os.makedirs(targetdir, 0o750)
- except os.error:
- raise stov_exceptions.DirectoryCreationFailedException()
- os.chdir(targetdir)
- if self.downloaded == 0:
- try:
- if self.__conf.outputlevel == "default":
- subprocess.check_call(["youtube-dl", "-f %s/%s"
- % (itag_value, video_codec), "-t",
- "http://www.youtube.com/watch?v=%s"
- % self.ytid],
- stderr=sys.stderr,
- stdout=open("/dev/null", "w"))
- elif self.__conf.outputlevel == "verbose":
- subprocess.check_call(["youtube-dl", "-f %s/%s"
- % (itag_value, video_codec), "-t",
- "http://www.youtube.com/watch?v=%s"
- % self.ytid],
- stderr=sys.stderr, stdout=sys.stdout)
- elif self.__conf.outputlevel == "quiet":
- subprocess.check_call(["youtube-dl", "-f %s/%s"
- % (itag_value, video_codec), "-t",
- "http://www.youtube.com/watch?v=%s"
- % self.ytid],
- stderr=open("/dev/null", "w"),
- stdout=open("/dev/null", "w"))
- except subprocess.CalledProcessError:
- self.failcnt = int(self.failcnt) + 1
- return False
- else:
- self.downloaded = 1
- return True
- def get_id(self):
- """Resturns the id attribute assigned to the object."""
- return self._id
|