ytainfo.py

#!/usr/bin/env python3
''' ytainfo - add channel info to archive database '''

import os
import sys
import argparse
import requests
import ytacommon as yta

# --------------------------------------------------------------------------- #
def addInfo(args):
    '''Add channel info to the archive database

    :param args: The command line arguments given by the user
    :type args: list
    '''
    parser = argparse.ArgumentParser(prog="ytainfo", description="Add channel info to existing archive databases")
    parser.add_argument("DIR", help="The directory containing the archive database to work on")
    args = parser.parse_args(args)

    path = os.path.normpath(os.path.abspath(args.DIR))
    dbPath = os.path.join(path, "archive.db")
    if not os.path.isdir(path) or not os.path.isfile(dbPath):
        parser.error("DIR must be a directory containing an archive database")

    #Check if database needs upgrade
    yta.upgradeDatabase(dbPath)

    add(dbPath)
# ########################################################################### #

# --------------------------------------------------------------------------- #
def add(dbPath):
    '''Add channel info to the archive database

    :param dbPath: The path of the archive database
    :type dbPath: string
    '''
    #Create/connect database
    db = createOrConnectDB(dbPath)
    print("ADDING CHANNEL INFO")
    #Get channel name
    while True:
        q = input("Channel name: ")
        if q:
            break
    name = q.strip()
    #Get channel url
    while True:
        q = input("Channel url: ")
        if q:
            break
    url = q.strip()
    #Get playlist name
    while True:
        q = input("Video playlist: ")
        if q:
            break
    playlist = q.strip()
    #Get channel language
    while True:
        q = input("Channel language code: ")
        if q:
            break
    language = q.strip()
    #Get channel Description
    print("Channel description (Press Ctrl-D [Ctrl-Z on WIN] directly to skip or after input to save):")
    desc = []
    while True:
        try:
            line = input()
            desc.append(line)
        except EOFError:
            break
    if desc:
        desc = '\n'.join(desc)
    else:
        desc = None
    #Get join date
    q = input("Join date (YYYY-MM-DD) or enter to skip: ")
    if q:
        joined = q.strip()
    else:
        joined = None
    #Get location
    q = input("Location (or enter to skip): ")
    if q:
        location = q.strip()
    else:
        location = None
    #Get links
    print("Add links: Prettyname first, then the url. Enter to continue")
    links = ""
    i = 0
    while True:
        i += 1
        q1 = input("Prettyname no. {} (or enter to continue): ".format(i))
        if not q1:
            break
        q2 = input("URL no. {}: ".format(i))
        if not q2:
            print("ERROR: URL must be specified")
            continue
        links += q1.strip() + '\t' + q2.strip() + '\n'
    if not links:
        links = None
    #Get profile picture
    while True:
        q = input("Profile picture URL (or enter to skip): ")
        if not q:
            profile = None
            profileformat = None
            break
        try:
            [profile, profileformat] = yta.loadImage(q.strip())
            break
        except requests.exceptions.HTTPError:
            print("ERROR: Invalid URL")
            continue
    #Get banner image
    while True:
        q = input("Banner image URL (or enter to skip): ")
        if not q:
            banner = None
            bannerformat = None
            break
        try:
            [banner, bannerformat] = yta.loadImage(q.strip())
            break
        except requests.exceptions.HTTPError:
            print("ERROR: Invalid URL")
            continue

    insert = "INSERT INTO channel(name, url, playlist, language, description, location, joined, links, profile, profileformat, banner, bannerformat, videos, lastupdate, dbversion, maxresolution, totalsize) VALUES(?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)"
    db.execute(insert, (name, url, playlist, language, desc, location, joined, links, profile, profileformat, banner, bannerformat, 0, 0, yta.__dbversion__, "default", 0))
    print("FINISHED ADDING CHANNEL INFO")

    yta.closeDB(db)
# ########################################################################### #

# --------------------------------------------------------------------------- #
def createEmpty(dbPath):
    '''Create a database without adding information

    :param dbPath: The path of the archive database
    :type dbPath: string
    '''
    #Create/connect database
    db = createOrConnectDB(dbPath)
    insert = "INSERT INTO channel(name, url, playlist, language, videos, lastupdate, dbversion, maxresolution, totalsize) VALUES(?,?,?,?,?,?,?,?,?)"
    db.execute(insert, ('', '', '', '', 0, 0, yta.__dbversion__, "default", 0))

    yta.closeDB(db)
# ########################################################################### #

# --------------------------------------------------------------------------- #
def createOrConnectDB(path):
    '''Create database with the required tables

    :param path: Path at which to store the new database
    :type path: string

    :raises: :class:``sqlite3.Error: Unable to create database

    :returns: Connection to the newly created database
    :rtype: sqlite3.Connection
    '''
    #Create database
    dbCon = yta.connectDB(path)
    #Create table
    yta.createChannelTable(dbCon)
    #Return database connection
    return dbCon
# ########################################################################### #

# --------------------------------------------------------------------------- #
if __name__ == "__main__":
    try:
        addInfo(sys.argv[1:])
    except KeyboardInterrupt:
        print("Aborted!")
# ########################################################################### #