Python - Manejo de archivos y listas Python

lista = [] #Lista vacia

pais=input("Elegir pais que desea actualizar") #Elegir el pais

habitantes=int(input("Ingresar la cantidad de nuevos habitantes"))#Cantidad de habitantes

pais= str()

habitantes=int()

archivo= open("registros.txt", "r")

linea= archivo.readlines()

for linea in archivo:

    if pais in linea:

        contenido += pais + "," + habitantes + "\n"

    else:

        contenido += linea

print(contenido)

archivo.close()

#-*- coding: utf -8 -*-

lista = [] #Lista vacia

pais=input("Elegir pais que desea actualizar: ") #Dejar espacio para que no se vea tan feo

habitantes=int(input("Ingresar la cantidad de nuevos habitantes: "))# igual

contenido = pais + ", "+ str(habitantes)

#pais= str()  #ya es una cadena

#habitantes=int()  #lo mismo, ya esta definido como entero

#archivo= open("registros.txt", "r") # esta forma es mala la mejor es 

try:

    with open ("registros.txt") as archivo:

       paises=archivo.read().splitlines()

        for cada in paises:

            if pais in cada:

                lista.append(contenido)

            else:

                lista.append(cada)

except FileNotFoundError:

    print("Error al intentar leer el archivo")

for x in lista:

    print(x)  #hasta acá sin sobreescribir el archivo

with open ("registros.txt","w") as archivo:  #guarda desde acá porque sobreescribe el archivo original

    for pais in lista:

        archivo.write(pais)

        archivo.write("\n")