Scripts/EPGScript.py at main · Kirvio/Scripts · 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
import logging
import time
import sys
import os
try:
    import paramiko
except ImportError as exc:
    print(exc)
    time.sleep(20)

"""
Скрипт закидывает с хоста (сначала на сервер,
потом в министру) готовый xmltv.xml файл
по ssh. Для передачи файлов используется
paramiko.Transport(sftp), для подключения
по ssh paramiko.SSHClient()
"""

remotepath = '/home/evandrushenko/xmltv.xml'
localpath = 'D:/EPG/xmltv.xml'

# Передача xmltv файла по sftp на сервер
def TransportData(frm, to, host='', port=22,\
                  SSHuser='', SSHsecret=''):
    result = False
    kak = os.path.isfile(localpath)
    if kak:
        try:
            transport = paramiko.Transport((host, port))
            transport.connect(username=SSHuser, password=SSHsecret)
            sftp = paramiko.SFTPClient.from_transport(transport)

            # Передача файлов
            sftp.put(frm, to)
        except (paramiko.SSHException, OSError) as err:
            print(err)
            time.sleep(20)
        else:
            sftp.close()
            transport.close()
            log.info("xml файл с EPG на сервере")
            result = True
        finally:
            return result
    else:
        print("xmltv файл отсутствует!")
        time.sleep(10)

# Соединение с сервером по ssh и копирование (sudo) с директории в директорию с министрой
def SSHConnect(SHhost='172.16.100.54', SHport=22,\
               SHuser='evandrushenko', SHpassword='1234567Andr'):
    try:
        ssh = paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        ssh.connect(hostname=SHhost, username=SHuser, \
                    password=SHpassword, port=SHport)

        chan = ssh.get_transport().open_session()
        chan.get_pty()

        cmd = 'sudo cp -L /home/evandrushenko/xmltv.xml /opt/docker_deploys/volumes/ministra_epg/_data/'
        chan.exec_command(cmd)
        chan.send(SHpassword + '\n')

        time.sleep(5)
    except (paramiko.SSHException, OSError) as err:
        print(err)
        time.sleep(20)
    else:
        ssh.close()
        log.info("Файл перенесён в докер")

if __name__ == '__main__':
    try:
        log = logging.getLogger(__name__)
        log.setLevel(logging.INFO)
        f_format = logging.Formatter('%(asctime)s %(levelname)s %(message)s')
        handler = logging.StreamHandler(sys.stdout)
        handler.setLevel(logging.INFO)
        handler.setFormatter(f_format)
        log.addHandler(handler)
    except Exception as exc:
        print(exc)
        time.sleep(20)
    else:
        ST = TransportData(frm=localpath, to=remotepath)
        time.sleep(2)
        if ST:
            SH = SSHConnect()
            time.sleep(2)
        else:
            log.info("Скрипт не выполнился")
            time.sleep(2)