import ipywidgets
from ipyleaflet import *

center = [44.4195, 38.2053]

s = Map(center=center, basemap=basemaps.Esri.WorldImagery, zoom=16)
marker = Marker(location=center, draggable=False)
s.add_layer(marker)

message1 = ipywidgets.HTML("h3Дворец Путина/h3")
popup = Popup(location=center, child=message1, close_button=False,\
    auto_close=False, close_on_escape_key=False)

s.add_layer(popup)
marker.popup = popup
display(s)

grep "Failed" /var/log/auth.log | grep "ssh" | grep -E -o "([0-9]{1,3}[\.]){3}[0-9]{1,3}" > ~/ips.txt

from geoip import geolite2

f = open("ips.txt").read().splitlines()
lookups = [geolite2.lookup(i) for i in set(f)]
locations = [i.location for i in lookups if i is not None]

m = Map(center=(0, 0), basemap=basemaps.OpenStreetMap.Mapnik, zoom=2)

heatmap = Heatmap(locations=locations, radius=5, min_opacity=0.2, blur=2)
m.add_layer(heatmap)
display(m);

lookups_ddos = [geolite2.lookup(i) for i in f]
locations_ddos = [i.location for i in lookups_ddos if i is not None]

m1 = Map(center=(0, 0), basemap = basemaps.Stamen.Terrain, zoom=2)

heatmap1 = Heatmap(locations=locations_ddos, radius=8, min_opacity=0.2, blur=2)

opacity_slider = ipywidgets.FloatSlider(description='Видимость:', min=0.01, max=0.5, value=0.2, step=0.01)
ipywidgets.jslink((opacity_slider, 'value'), (heatmap1, 'min_opacity'))
widget_control1 = WidgetControl(widget=opacity_slider, position='topright')
m1.add_control(widget_control1)
m1.add_layer(heatmap1)

display(m1);

import pandas as pd

countries = {}

for ip in lookups:
    if ip == None:
        continue
    if not ip.country in countries.keys():
        countries[ip.country] = 1
    else:
        countries[ip.country] += 1

countries_rank = sorted(countries.keys(), key=lambda id: countries.get(id))
countries_loc = pd.read_csv("countries.csv", sep=" \t", engine="python")

m2 = Map(center=(0, 0), basemap = basemaps.Stamen.Toner, zoom=3)

for country in countries_rank:
    query_number = countries.get(country)
    info = countries_loc[countries_loc["country"] == country].values[0]
    text = "{1}: {0}".format(query_number, country)

    popup = Popup(location=(info[1], info[2]), child=ipywidgets.Label(text),\
                  close_button=False, auto_close=False, close_on_escape_key=False)
    m2.add_layer(popup)

display(m2);