Geolocation_save/save_geolocation.py at master · Kjeck/Geolocation_save · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
# -*- coding: utf-8 -*-
# !/usr/bin/python3.10
''' Saves geolocation on the map to html file & creat QR-code '''

__version__ = '1.1'

from time import sleep
from termcolor import cprint
from pyfiglet import Figlet
import folium
import os
import colorama
import qrcode

# Enable AHCI support for color text in the console when compiling in win app
colorama.init()


def save_location_qr(lat, lon):
    data = f'https://maps.google.com/local?q={lat},{lon}'
    file_name = f'location_{lat}_{lon}.png'
    img = qrcode.make(data)
    try:
        img.save(file_name)
        cprint('QR-code save to:     ' + file_name, 'yellow')
        print()
    except:
        cprint('\nНе удалось сохранить QR-code\n', 'red')
        print()


def save_location_html(lat, lon):
    try:
        coordinates = (lat, lon)
        file_name = f'location_{lat}_{lon}.html'

        map = folium.Map(location=coordinates, zoom_start=12)
        folium.Marker(location=coordinates,
                      icon=folium.Icon(color='red')).add_to(map)
        map.save(file_name)
        cprint('Geolocation save to: ' + file_name, 'yellow')

        save_location_qr(lat, lon)
        try:
            os.startfile(file_name)
        except:
            pass
    except:
        cprint('\nНе удалось сохранить локацию в файл\n', 'red')


def input_location():
    loop_run = True
    while loop_run:
        cprint('Введите через пробел Широту Долготу (<Q> для выхода)', 'green')
        location = input('> ')
        if location[:1] not in ['Q', 'q', 'Й', 'й']:
            if location.find(' ') > 0:
                try:
                    lat, lon = location.split()
                except:
                    cprint('Не верный ввод координат!', 'red')
                    input_location()

                lat = lat.replace(',', '.')
                lon = lon.replace(',', '.')
                save_location_html(lat, lon)
            else:
                cprint('Введите координаты через пробел!', 'red')
        else:
            loop_run = False


def print_info(type='copyright', font='slant', color='yellow'):

    if type == 'head':
        text = 'SAVE LACATION ON THE MAP'
        f = Figlet(font=font)
        cprint(f.renderText(text), color)
    elif type == 'copyright':
        cprint('\nSee you later ...', color)
        print()
        cprint('© Evgeniy Kotelnikov (Kjeck)  ', 'white', 'on_cyan')
        cprint('  https://github.com/Kjeck    \n', 'white', 'on_cyan')


def main():
    os.system('CLS')
    print_info('head')
    input_location()
    print_info()
    # For compilations to win app
    sleep(3)


if __name__ == '__main__':
    main()