mirror of
https://github.com/knejadshamsi/zele-utils.git
synced 2024-11-14 17:40:28 -05:00
53 lines
2.0 KiB
Python
53 lines
2.0 KiB
Python
from bs4 import BeautifulSoup
|
|
import pandas,pyproj, re
|
|
from shapely.geometry import Point
|
|
|
|
def camel_to_snake(name):
|
|
return re.sub(r'(?<!^)(?=[A-Z])', '_', name).lower()
|
|
|
|
def process_buffer_population(data):
|
|
transformer = pyproj.Transformer.from_crs('EPSG:2950', 'EPSG:4326', always_xy=True)
|
|
PERSON_LIST = []
|
|
elements = BeautifulSoup(data, 'lxml-xml')
|
|
for person in elements.find_all('person'):
|
|
person_obj = {}
|
|
person_obj['id'] = person['id']
|
|
activity = person.find('plan').find('activity')
|
|
lat, lon = transformer.transform(activity['x'], activity['y'])
|
|
person_obj['coordinates'] = Point(lon,lat)
|
|
person_obj['time'] = activity['end_time']
|
|
for attr in person.find_all('attribute'):
|
|
person_obj[camel_to_snake(attr['name'])] = attr.get_text()
|
|
PERSON_LIST.append(person_obj)
|
|
return pandas.DataFrame(PERSON_LIST)
|
|
|
|
def process_travels(data):
|
|
transformer = pyproj.Transformer.from_crs('EPSG:2950', 'EPSG:4326', always_xy=True)
|
|
activities_list = []
|
|
elements = BeautifulSoup(data, 'lxml-xml')
|
|
for person in elements.find_all('person'):
|
|
person_id = person['id']
|
|
plan = person.find('plan')
|
|
|
|
for i, activity in enumerate(plan.find_all('activity'), start=1):
|
|
activity_obj = {
|
|
'id': person_id,
|
|
'activity_order': i,
|
|
'activity_type': activity.get('type'),
|
|
'facility': activity.get('facility'),
|
|
'start_time': activity.get('start_time') or None,
|
|
'end_time': activity.get('end_time') or None
|
|
}
|
|
|
|
if 'x' in activity.attrs and 'y' in activity.attrs:
|
|
lon, lat = transformer.transform(float(activity['x']), float(activity['y']))
|
|
activity_obj['coordinates'] = Point(lon, lat)
|
|
else:
|
|
activity_obj['coordinates'] = None
|
|
|
|
activities_list.append(activity_obj)
|
|
|
|
return pandas.DataFrame(activities_list)
|
|
|
|
|