#!python
#
# Create a header file for extracted Wikipedia text.

import os
import argparse
from wikivector import wiki


def main(wiki_text_dir):
    df = wiki.dump_pages(wiki_text_dir)
    header_file = os.path.join(wiki_text_dir, 'header.csv')
    df.to_csv(header_file, index=False)


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('wiki_dir', help="Path to output from WikiExtractor")
    args = parser.parse_args()

    main(args.wiki_dir)
