import telnetlib
import os
address = '130.240.134.5'
def saveRouterInformation(address, filename):
    tn = telnetlib.Telnet(address, '3034')
    tn.read_until('Username:')
    tn.write(b'natverklab\n')
    tn.read_until('Password:')
    tn.write(b'cisco\n')            #Loggar in
    tn.read_until('#')
    tn.write(b'show version\n\n')     #Tar fram versionen och sparar den i filen
    version = tn.read_until('#')
    tn.write('show ip route\n')
    iproute = tn.read_until('#')
    f = open(filename, 'w')
    f.write(version)                #Sparar show version och show ip route i en fil
    f.write(iproute)
    f.close()
    f = open(filename, 'r')              #Gör filen läsbar för att plocka ut info och spara om
    f.readline() + f.readline()
    f.read(38)
    ios = f.read(16)

    i = 0
    while i < 7:
        f.readline()
        i+=1
    hostname = f.readline(4)

    f.readline() + f.readline()
    f.readline(21)
    img = f.readline(31)

    f.readline() + f.readline()
    ver = f.readline(10)
    f.readline(42)
    mem = f.readline()
    i = 0
    while i < 19:
        f.readline()
        i = i+1
    i=0
    ipr =''

    while i < 5:
        ipr += f.readline()
        i+=1
    f.close()           #Stänger filen

    f = open(filename, 'w')  #Öppnar filen igen för att kunna spara över med ny data
    f.write('Version:       '+ ver+'\nHostname:     '+hostname+'\n')
    f.write('Img-fil:       '+img+'\nIOS-version:     '+ios+'\nMinne:       '+mem+'\nRouting tabell:\n'+ipr)
    f.close()
    tn.close()