Python подключиться к удаленному хосту и захватить веб-страницу на локальном хосте - PullRequest
Новая
       47

Python подключиться к удаленному хосту и захватить веб-страницу на локальном хосте

0 голосов
/ 28 апреля 2018

Я посмотрел здесь на многопроцессорность, по большей части я его сузил, но последний бит почему-то не работает.

Context

У меня есть удаленный веб-сервер, к которому я подключаюсь, у меня есть порт, который перенаправляет страницу HTTP на локальный порт. Мне нужно подключиться к удаленному хосту, пока я подключен, мне нужно открыть страницу HTTP на моем локальном компьютере и захватить страницу.

Код 

from selenium import webdriver
from pyvirtualdisplay import Display
import paramiko
import multiprocessing
import time

def grabscreendisplay():
 time.sleep(10)
 print('running display code now... ')
 display = Display(size=(1024, 768), visible=0)
 display.start()
 driver=webdriver.Chrome('/usr/local/bin/chromedriver')
 URL="http://localhost:9012"
 driver.get(URL)
 screenshot=driver.save_screenshot('my_screenshot.png')
 driver.quit()


def getserver():
    SERV = raw_input('Please enter the server you would like to connect to: ')
    return SERV
def connect_to_server(SERV):
    print(SERV)
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    print('Connecting to ' + SERV)
    ssh.connect(SERV, username='root', key_filename='...')
    connected = True
    print('Connected to ' + SERV)
    time.sleep(30)

def main():
    SERV = getserver()
    p1 = multiprocessing.Process(name='p1', target=connect_to_server(SERV))
    p2 = multiprocessing.Process(name='p2', target=grabscreendisplay())
    p2.start()
    p1.start()

if __name__ == "__main__":
    main()

Проблема возникла

.png просто отображает сбой подключения к порту («localhost отказался подключаться»)

1 Ответ

0 голосов
/ 29 апреля 2018

SSHTunnel

Поэтому мне пришлось исследовать туннель SSH, который создает случайно сгенерированный локальный порт связывания для сопоставления с портом удаленного назначения.

Вскоре после повторных попыток мне удалось получить конечный результат, однако, если вы ищете ответ на проблему, подобную моей, я предоставлю свой код, а также пример, который надеется подойти всем вместо моей конкретной проблемы 

from sshtunnel import SSHTunnelForwarder
import paramiko
import time
SSHTunnelForwarder.SSH_TIMEOUT = SSHTunnelForwarder.TUNNEL_TIMEOUT = 5.0

                     ### Setting up the SSHTunnel ###

with SSHTunnelForwarder(
        SERV, #IP of 10.10.10.1
        ssh_username='admin',
        ssh_pkey="...",
        remote_bind_address=(SERV, 80), # map the 10.10.10.1 port 80
        ) as tunnel: 

                     ### preliminary SSH connection ###

        client = paramiko.SSHClient() # set up SSH connection
        client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        client.connect(SERV, username='admin', key_filename='...')

             ### Setting up the pyvirtualdisplay on local machine ###

        display = Display(size=(1024, 768), visible=0) # set virtual window size
        display.start()
        driver=webdriver.Chrome('/usr/local/bin/chromedriver') # I'm using Chromedriver
        print(tunnel.local_bind_port) # this prints the randomly generated local port, for example in my case it would be localhost:32842 and would display the web page of the remote host
        time.sleep(5)                    
        URL='http://localhost'+':'+str(tunnel.local_bind_port)
        driver.get(URL)
        time.sleep(10)
        print(driver.title)
        screenshot=driver.get_screenshot_as_file('newscreen.png')

упрощенный 

 with SSHTunnelForwarder(
        <REMOTE_HOST IP>,
        ssh_username=<ssh_username>,
        ssh_pkey=<location of own private key>,
        remote_bind_address=(<REMOTE_HOST>, <REMOTE_PORT>),
        ) as tunnel: 
        client = paramiko.SSHClient() # set up SSH connection
        client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        client.connect(<REMOTE_IP>, username=<ssh_username>, key_filename=<location of own private key>)
        display = Display(size=(1024, 768), visible=0)
        display.start()
        driver=webdriver.Chrome('/usr/local/bin/chromedriver')
        print(tunnel.local_bind_port)
        while True:
           sleep(30)
           # ctrl + c to stop program
        # Go into own local web browser and enter http://localhost:<local_bind_port> and you should see the web page

Сайт для Chromedriver

...