summaryrefslogtreecommitdiff
path: root/to_csv.py
blob: e08bc24fa4389ac38ea6429d432ffbf480711b71 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#!/usr/bin/env python3
import csv
from mgs import MGSPiracy
from argparse import ArgumentParser

if __name__ == '__main__':
    # parse arguments
    argp = ArgumentParser()
    argp.add_argument('--output', type=str, default='output.csv', help='CSV output file name')
    argp.add_argument('--from', type=int, default=0, help='First page', dest='_from')
    argp.add_argument('--to', type=int, default=10, help='Last page')
    args = argp.parse_args()

    # get cases
    mgs = MGSPiracy(from_page=args._from, to_page=args.to)
    cases = mgs.get_cases()

    # write to csv
    f = open(args.output, 'w', newline='')
    csv_writer = csv.writer(f)

    for case in cases:
        csv_writer.writerow((
            case['date'],
            case['statement_number'],
            case['applicant'],
            case['object'],
            case['doc_link'],
            case['violation_links']))

    f.close()