mirror of https://github.com/MISP/misp-dashboard
79 lines
2.1 KiB
Python
Executable File
79 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python3.5
|
|
import json
|
|
import math
|
|
import os
|
|
import shlex
|
|
import shutil
|
|
import sys
|
|
import time
|
|
from subprocess import PIPE, Popen
|
|
|
|
import redis
|
|
import requests
|
|
|
|
URL_OPEN_MAP = "http://tile.openstreetmap.org/{zoom}/{x}/{y}.png"
|
|
MAP_DIR = "static/maps/"
|
|
ZOOM = 17
|
|
dlat = 2
|
|
dlon = 2
|
|
serv = redis.StrictRedis('localhost', 6250, 2)
|
|
channel_proc = "CoordToProcess"
|
|
channel_disp = "PicToDisplay"
|
|
|
|
|
|
def lon2tile(lon, zoom):
|
|
return (math.floor((lon+180)/360*math.pow(2,zoom)))
|
|
|
|
def lat2tile(lat, zoom):
|
|
return (math.floor((1-math.log(math.tan(lat*math.pi/180) + 1/math.cos(lat*math.pi/180))/math.pi)/2 *math.pow(2,zoom)))
|
|
|
|
|
|
def main():
|
|
pubsub = serv.pubsub(ignore_subscribe_messages=True)
|
|
pubsub.subscribe(channel_proc)
|
|
|
|
while True:
|
|
for d in pubsub.listen():
|
|
#data = json.loads(d.decode('utf8'))
|
|
download_maps(d['data'])
|
|
|
|
def create_box_around_coord(lon,lat):
|
|
pics = []
|
|
for dx in range(-1,2,1):
|
|
for dy in range(-1,2,1):
|
|
pics.append(URL_OPEN_MAP.format(x=lon+dx, y=lat+dy, zoom=ZOOM))
|
|
return pics
|
|
|
|
def download_and_merge(url_list, coord):
|
|
to_process = b""
|
|
for i, url in enumerate(url_list):
|
|
to_process += (str(i+1)+' '+url+'\n').encode('utf8')
|
|
|
|
map_name = "map_{lon}-{lat}.png".format(lon=coord['lon'], lat=coord['lat'])
|
|
path = os.path.join(MAP_DIR, map_name)
|
|
cmd1 = "parallel --colsep ' ' wget -O maps/{1} {2}"
|
|
cmd2 = "montage -geometry +0+0 maps/1 maps/4 maps/7 maps/2 maps/5 maps/8 maps/3 maps/6 maps/9 "+path
|
|
|
|
# Donwload tiles
|
|
p = Popen(shlex.split(cmd1), stdout=PIPE, stdin=PIPE)
|
|
p.communicate(input=to_process)
|
|
|
|
# Combine tiles
|
|
p = Popen(shlex.split(cmd2), stdout=PIPE, stdin=PIPE)
|
|
|
|
return map_name
|
|
|
|
def download_maps(coord):
|
|
coord = json.loads(coord.decode('utf8'))
|
|
print(str(coord))
|
|
lat = lat2tile(coord['lat'], ZOOM)
|
|
lon = lon2tile(coord['lon'], ZOOM)
|
|
|
|
urls = create_box_around_coord(lon, lat)
|
|
map_name = download_and_merge(urls, coord)
|
|
print(map_name)
|
|
serv.publish(channel_disp, json.dumps({ "path": map_name, "coord": coord }))
|
|
|
|
if __name__ == '__main__':
|
|
main()
|