gpx2xlsx/gpx2xlsx.py at master · Planetcell/gpx2xlsx · 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
# -*- coding: utf-8 -*-
from xml.dom.minidom import parse
import xml.dom.minidom
import xlsxwriter
import os

class GpxHandler():
    def __init__(self, pathin ,pathout):
        '''
        初始化 init
        :param pathin: 需要解析的gpx文件所在目录 directory in which the GPX files are located
        :param pathout: 输出xlsx文件的目录 directory to output the XLSX files
        '''
        self.pathin = pathin
        self.pathout = pathout

    def generate(self, infile, outfile):
        # 解析gpx文件 parse the GPX file
        DOMTree = xml.dom.minidom.parse(infile)
        gpx = DOMTree.documentElement
        trk = gpx.getElementsByTagName('trk')[0]
        trkseg = trk.getElementsByTagName('trkseg')[0]
        trkpts = trkseg.getElementsByTagName('trkpt')
        # 输出xlsx文件 output the XLSX file
        wbk = xlsxwriter.Workbook(outfile)
        sheet = wbk.add_worksheet('sheet 1')
        n = 0
        sheet.write(n, 0, 'lat')
        sheet.write(n, 1, 'lon')
        sheet.write(n, 2, 'ele')
        sheet.write(n, 3, 'time')
        n = n + 1
        for trkpt in trkpts:
            lat = trkpt.getAttribute('lat')
            lon = trkpt.getAttribute('lon')
            ele = trkpt.getElementsByTagName('ele')[0].firstChild.data
            time = trkpt.getElementsByTagName('time')[0].firstChild.data
            sheet.write(n, 0, lat)
            sheet.write(n, 1, lon)
            sheet.write(n, 2, ele)
            sheet.write(n, 3, time)
            n = n + 1
        wbk.close()

    def dealGpx(self):
        '''
        处理 deal with the GPX files
        '''
        if not os.path.isdir(self.pathout):
            os.makedirs(self.pathout)
        for root, dirs, files in os.walk(self.pathin):
            for file in files:
                # 判断是否为gpx文件 judge whether it is a GPX file
                if os.path.splitext(file)[1] == '.gpx':
                    filename = os.path.splitext(file)[0]
                    self.generate(self.pathin + '\\' + file, self.pathout + '\\' + filename + r'.xlsx')
        print("Figured out.")

if __name__ == '__main__':
    pathin = "pathin"
    pathout = "pathout"
    gpxHandler = GpxHandler(pathin, pathout)
    gpxHandler.dealGpx()