61 lines
1.3 KiB
Python
61 lines
1.3 KiB
Python
import csv
|
|
import shapely
|
|
|
|
from nexrad.db import DatabaseTable
|
|
from nexrad.coord import COORD_SYSTEM
|
|
|
|
class WFO(DatabaseTable):
|
|
__slots__ = (
|
|
'code', 'city', 'state', 'address', 'coord'
|
|
)
|
|
|
|
__table__ = 'nexrad_wfo'
|
|
__key__ = 'code'
|
|
|
|
__columns__ = (
|
|
'code', 'city', 'state', 'address', 'coord'
|
|
)
|
|
|
|
__columns_read__ = {
|
|
'coord': 'ST_AsText(coord) as coord'
|
|
}
|
|
|
|
__values_read__ = {
|
|
'coord': shapely.from_wkt
|
|
}
|
|
|
|
__columns_write__ = {
|
|
'coord': 'MakePoint(:coord_lon, :coord_lat, {crs})'.format(crs=COORD_SYSTEM)
|
|
}
|
|
|
|
__values_write__ = {
|
|
'coord': lambda v: {'coord_lon': v.x, 'coord_lat': v.y}
|
|
}
|
|
|
|
id: int
|
|
code: str
|
|
city: str
|
|
state: str
|
|
adadress: str
|
|
|
|
@staticmethod
|
|
def from_tsv_row(row: list):
|
|
wfo = WFO()
|
|
wfo.code = row[2]
|
|
wfo.city = row[0]
|
|
wfo.state = row[1]
|
|
wfo.address = row[3]
|
|
wfo.coord = shapely.Point(float(row[5]), float(row[4]))
|
|
|
|
return wfo
|
|
|
|
@staticmethod
|
|
def each_from_tsv(file: str):
|
|
with open(file) as fh:
|
|
reader = csv.reader(fh, delimiter='\t')
|
|
|
|
for row in reader:
|
|
for i in range(0, len(row)):
|
|
row[i] = row[i].rstrip()
|
|
|
|
yield WFO.from_tsv_row(row)
|