PDF-Extractor/test.py at master · trkhoi/PDF-Extractor · GitHub

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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
import pdftotext

CONFIG = {
        'From': {
            'row': [7, 8],
            'column': [29, 91],
            'startFlex': False,
            'endObject': None,
            'verticalFlex': False
        },
        'To': {
            'row': [7, 8],
            'column': [113, None],
            'startFlex': False,
            'endObject': None,
            'verticalFlex': False
        },
        'Booking No.': {
            'row': [9, 10],
            'column': [29, 91],
            'startFlex': False,
            'endObject': None,
            'verticalFlex': False
        },
        'Date': {
            'row': [8, 9],
            'column': [29, 91],
            'startFlex': False,
            'endObject': None,
            'verticalFlex': False
        },
        'Shipper Name & Address': {
            'row': [12, 18],
            'column': [4, 91],
            'startFlex': False,
            'endObject': 'Notify Party',
            'verticalFlex': True
        },
        'Consignee Name & Address': {
            'row': [12, 18],
            'column': [92, None],
            'startFlex': False,
            'endObject': 'Port of Loading',
            'verticalFlex': True
        }
}

CURR_CONFIG = {}
fileName = "VN101466/SI_HANV07496600.pdf"

if __name__ == '__main__':
    with open(fileName, "rb") as f:
        pdf = pdftotext.PDF(f)
    page = pdf[0].split("\n")

    # for i in range(len(page)):
    #     print(i)
    #     print([str(k) + page[i][k] for k in range(len(page[i]))])
    data = {}
    for key in CONFIG:
        # print(key)
        row = CONFIG[key]['row']
        column = CONFIG[key]['column']
        # print(row)
        # print(column)
        lines = page[row[0]:row[1]]
        data[key] = '\n'.join([x[column[0]:column[1]] for x in lines])

    for key in data:
        print("%s: %s" % (key, data[key]))