Merge pull request #17 from zuzhi/master

add `csv` CLI tool

Author: wandergis

README.md

@@ -1,21 +1,48 @@
 # 坐标转换模块
-此模块用于百度坐标系(bd-09)、火星坐标系(国测局坐标系、gcj02)、WGS84坐标系的相互转换，并提供中文地址到坐标的转换功能，仅使用Python标准模块，无其他依赖。中文地址到坐标转换使用高德地图API，需要[申请](http://lbs.amap.com/)API KEY。
+
+此模块用于百度坐标系(bd-09)、火星坐标系(国测局坐标系、gcj02)、WGS84坐标系的相互转换，并提供中文地址到坐标的转换功能，仅使用Python标准模块，无其他依赖。
+
+中文地址到坐标转换使用高德地图API，需要[申请](http://lbs.amap.com/)API KEY。
 
 需要js版本可以移步[coordtransform](https://github.com/wandergis/coordtransform)
 
-# 使用说明
+## 使用说明
+
+### 方法说明
+
+```bash
+# 方法说明
+gcj02_to_bd09(lng, lat) # 火星坐标系->百度坐标系
+bd09_to_gcj02(lng, lat) # 百度坐标系->火星坐标系
+wgs84_to_gcj02(lng, lat) # WGS84坐标系->火星坐标系
+gcj02_to_wgs84(lng, lat) # 火星坐标系->WGS84坐标系
+bd09_to_wgs84(lng, lat) # 百度坐标系->WGS84坐标系
+wgs84_to_bd09(lng, lat) # WGS84坐标系->百度坐标系
+
+# 中文地址到火星坐标系, 需要高德地图API Key
+g = Geocoding('API_KEY')  # 这里填写你的高德Api_Key
+g.geocode('北京市朝阳区朝阳公园')
+```
+
+### 测试
+
+```bash
+# 测试
+$ python coordTransform_utils.py
+[128.54944656269413, 37.07113427883019] [128.5365893261212, 37.058754503281534] [128.54820547949757, 37.065651049489816] [128.53779452050244, 37.06434895051018] [128.53136876750008, 37.0580926428705] [128.55468192918485, 37.07168344938498] None
 ```
-    lng = 128.543
-    lat = 37.065
-    result1 = gcj02_to_bd09(lng, lat)#火星坐标系->百度坐标系
-    result2 = bd09_to_gcj02(lng, lat)#百度坐标系->火星坐标系
-    result3 = wgs84_to_gcj02(lng, lat)#WGS84坐标系->火星坐标系
-    result4 = gcj02_to_wgs84(lng, lat)#火星坐标系->WGS84坐标系
-    result5 = bd09_to_wgs84(lng, lat)#百度坐标系->WGS84坐标系
-    result6 = wgs84_to_bd09(lng, lat)#WGS84坐标系->百度坐标系
-
-	#中文地址到火星坐标系,需要高德地图API Key
-    g = Geocoding('API_KEY')  # 这里填写你的高德Api_Key
-    result7 = g.geocode('北京市朝阳区朝阳公园')
-    print result1, result2, result3, result4, result5, result6, result7
+
+## 批量转换csv文件(coord_converter.py)
+
+使用方法：
+
+```bash
+$ python coord_converter.py -h
+
+Usage: coord_converter.py -i <input> -o <output> -t <type>
+
+where <type> is one of:
+    g2b, b2g, w2g, g2w, b2w, w2b
+
+Example: coord_converter.py -i /path/to/input_file.csv -o /path/to/output_file.csv -t b2g
 ```

coord_converter.py

@@ -0,0 +1,81 @@
+#!/usr/bin/python
+
+import csv
+import sys
+import getopt
+from coordTransform_utils import gcj02_to_bd09
+from coordTransform_utils import bd09_to_gcj02
+from coordTransform_utils import wgs84_to_gcj02
+from coordTransform_utils import gcj02_to_wgs84
+from coordTransform_utils import bd09_to_wgs84
+from coordTransform_utils import wgs84_to_bd09
+
+# Configuation
+# Input file name
+INPUT = ''
+# Output file name
+OUTPUT = ''
+# Convert type: g2b, b2g, w2g, g2w, b2w, w2b
+TYPE = ''
+
+def convert():
+    with open(INPUT, 'r') as input_file:
+        input_file_reader = csv.reader(input_file)
+        results = []
+        for lat, lng in input_file_reader:
+            result = convert_by_type(float(lng), float(lat), TYPE)
+            results.append(result)
+
+    with open(OUTPUT, 'w') as output_file:
+        output_file_writer = csv.writer(output_file)
+        for result in results:
+            output_file_writer.writerow(result)
+
+def convert_by_type(lng, lat, type):
+    if type == 'g2b':
+        return gcj02_to_bd09(lng, lat)
+    elif type == 'b2g':
+        return bd09_to_gcj02(lng, lat)
+    elif type == 'w2g':
+        return wgs84_to_gcj02(lng, lat)
+    elif type == 'g2w':
+        return gcj02_to_wgs84(lng, lat)
+    elif type == 'b2w':
+        return bd09_to_wgs84(lng, lat)
+    elif type == 'w2b':
+        return wgs84_to_bd09(lng, lat)
+    else:
+        print('Usage: The <type> must be in one of g2b, b2g, w2g, g2w, b2w, w2b')
+        sys.exit()
+
+def usage():
+    print('Usage: coord_converter.py -i <input> -o <output> -t <type>')
+    print('')
+    print('where <type> is one of:')
+    print('    g2b, b2g, w2g, g2w, b2w, w2b')
+    print('')
+    print('Example: coord_converter.py -i /path/to/input_file.csv -o /path/to/output_file.csv -t b2g')
+
+if __name__ == "__main__":
+    argv = sys.argv[1:]
+    try:
+        opts, args = getopt.getopt(argv, "hi:o:t:", ["input=", "output=", "type="])
+    except getopt.GetoptError:
+        usage()
+        sys.exit(2)
+    for opt, arg in opts:
+        if opt == '-h':
+            usage()
+            sys.exit()
+        elif opt in ("-i", "--input"):
+            INPUT = arg
+        elif opt in ("-o", "--output"):
+            OUTPUT = arg
+        elif opt in ("-t", "--type"):
+            TYPE = arg
+
+    if not (INPUT and OUTPUT and TYPE):
+        usage()
+        sys.exit()
+
+    convert()