run.py

import os
import random
import subprocess

import config
import twitter

# config
script_dir = os.path.dirname(os.path.abspath(__file__))

try:
    cfg = config.Config(os.path.join(script_dir,'settings.cfg'))
except config.ConfigFormatError:
    print('Error: Check settings.cfg')
    exit()

# settings
directory = cfg['settings.general.directory']
video_directory = cfg['settings.general.video.directory']
clip_length = cfg['settings.general.video.length']
video_chance = cfg['settings.general.video.chance']

# twitter keys
CONSUMER_KEY_INPUT = cfg['settings.keys.consumer.key']
CONSUMER_SECRET_INPUT = cfg['settings.keys.consumer.secret']
ACCESS_TOKEN_KEY_INPUT = cfg['settings.keys.access.key']
ACCESS_TOKEN_SECRET_INPUT = cfg['settings.keys.access.secret']

# etc.
tmpfile_img = os.path.join(script_dir,cfg['settings.etc.tmpfile.img'])
tmpfile_vid = os.path.join(script_dir,cfg['settings.etc.tmpfile.vid'])


# randomly select an mkv video from input directory
# look at help diagram to understand which use cases are supported
def get_random_video_filepath(directory):
    outdir = ''
    direc = directory
    while os.path.isdir(direc):
        outdir += random.choice(os.listdir(direc))
        if outdir.endswith(('mkv', 'mp4', 'avi')):
            break
        else:
            outdir += '/'
            direc += outdir
    return outdir


# use ffmpeg to generate screenshot at timestamp
def generate_random_screenshot_locally(filepath):
    random_time = random.uniform(0.00, get_length(filepath))
    command_img = [
        'ffmpeg', '-y',
        '-ss', str(random_time),
        '-i', filepath,
        '-vframes', '1',
        '-vf', 'scale=1920:-1',
        '-q:v', '1',
        '-qmin', '1',
        tmpfile_img
    ]
    subprocess.call(command_img)
    return tmpfile_img


# use ffmpeg to generate 5 sec clip at timestamp
def generate_random_clip_locally(filepath):
    random_time = random.uniform(0.00, get_length(filepath) - float(clip_length))
    command_vid = [
        'ffmpeg', '-y',
        '-ss', str(random_time),
        '-i', filepath,
        '-t', str(clip_length),
        '-ac', '2',
        '-sn',
        '-map_chapters', '-1',
        '-c:v', 'libx264',
        '-c:a', 'aac',
        '-vf', 'scale=1280:-1',
        '-pix_fmt', 'yuv420p',
        '-movflags', '+faststart',
        tmpfile_vid
    ]
    subprocess.call(command_vid)
    return tmpfile_vid


# use ffprobe to get the length of the mkv
def get_length(filepath):
    command_info = [
        'ffprobe',
        '-i', filepath,
        '-show_entries', 'format=duration',
        '-v', 'quiet',
        '-of', 'csv=%s' % 'p=0'
    ]
    duration = subprocess.check_output(command_info)
    return float(duration)


# uploads the screenshot/clip to twitter
def post_update(tmpfile):
    try:
        api = twitter.Api(consumer_key=CONSUMER_KEY_INPUT,
                        consumer_secret=CONSUMER_SECRET_INPUT,
                        access_token_key=ACCESS_TOKEN_KEY_INPUT,
                        access_token_secret=ACCESS_TOKEN_SECRET_INPUT)
        return api.PostUpdate('', tmpfile)
    except twitter.error.TwitterError:
        print('Error: Invalid key')
    


# checks if we should generate a screenshot or clip
def check_video():
    r = random.random()
    if r <= video_chance:
        return True
    else:
        return False


if __name__ == '__main__':

	 # determine if we should generate a video or screenshot
    shouldGenerateVideo = check_video()

	# if we are generating a video, set our working directory to the video directory if one exists.
    if shouldGenerateVideo and video_directory:
        directory = video_directory

    # find our random video by parsing directory
    try:
        filepath = directory + get_random_video_filepath(directory)
    except IndexError:
        print('Error: No video file')
        exit()

    # generate output file
    try:
        if shouldGenerateVideo:
            output = generate_random_clip_locally(filepath)
        else:
            output = generate_random_screenshot_locally(filepath)
    except subprocess.CalledProcessError:
        print('Error: Invalid file path')
        exit()

    # post to twitter
    post_update(output)