24 lines
451 B
Python
24 lines
451 B
Python
|
import re
|
||
|
|
import argparse
|
||
|
|
import os
|
||
|
|
import pandas as pd
|
||
|
|
|
||
|
|
SCRIPT_DIR = os.path.abspath(os.path.dirname(__file__))
|
||
|
|
IN_DATASET = os.path.join(SCRIPT_DIR, "data.csv")
|
||
|
|
|
||
|
|
|
||
|
|
def search(query):
|
||
|
|
df = pd.read_csv(IN_DATASET)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument("query", help="the query to search the corpus with", type=str)
|
||
|
|
args = parser.parse_args()
|
||
|
|
search(query)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|