import duckdb

create_docs = """ CREATE TABLE docs (
    trecid          VARCHAR,
    luceneid        INTEGER,
    length          VARCHAR,
    irr             INTEGER
);
"""

create_dict = """ CREATE TABLE dict (
    termid          INTEGER,
    term            VARCHAR,
    df              INTEGER
);
"""

create_terms = """CREATE TABLE terms (
    termid          INTEGER,
    luceneid           INTEGER,
    count           INTEGER
);
"""

copy_docs = """ COPY docs
FROM '/Users/Daan/IdeaProjects/docs.csv'
WITH (DELIMITER '|')
"""

copy_dict = """ COPY dict
FROM '/Users/Daan/IdeaProjects/dict.csv'
WITH (DELIMITER '|')
"""

copy_terms = """ COPY terms
FROM '/Users/Daan/IdeaProjects/terms.csv'
WITH (DELIMITER '|')
"""

test_statement = "INSERT INTO terms VALUES (10, 20, 30);"

cursor = duckdb.connect(':memory:').cursor()
cursor.execute(create_dict)
cursor.execute(create_docs)
cursor.execute(create_terms)
cursor.execute(copy_docs)
cursor.execute(copy_dict)
cursor.execute(copy_terms)
cursor.execute("ALTER TABLE docs DROP irr")
print(cursor.execute("SELECT * FROM docs WHERE luceneid = 595036").fetchall())