示例#1
0
def train():
    parser = argparse.ArgumentParser()
    parser.add_argument("--db", help="Filename for sqlite3 database")
    parser.add_argument("dir", help="Input directory")

    # Get arguments or set to defaults
    args = parser.parse_args()
    database = args.db or DEFAULT_DATABASE
    directory = args.dir

    # Connect to the database
    session = connect_to_database(database)
    rhyme_dict = RhymeDictionary(session, 2)

    print("Training...")
示例#2
0
def generate_song():
    parser = argparse.ArgumentParser()
    parser.add_argument("--db", help="Filename for sqlite3 database")
    parser.add_argument("file", help="Output file")

    # Get arguments or set to defaults
    args = parser.parse_args()
    database = args.db or DEFAULT_DATABASE
    filename = args.file

    # Connect to the database
    session = connect_to_database(database)
    rhyme_dict = RhymeDictionary(session, 2)

    print("Generating a song...")
    with open(filename, "wb") as output_file:
        output_file.write(rhyme_dict.random_rhyme("lips"))
示例#3
0
def get_rhyme_dict():
    Session = connect_to_database("test.db")
    return RhymeDictionary(Session, 0)