modificar albums

#função de modificar
    def modificar():
        print('\n           -MODIFICÃO-            ')
        cursor=Base.connect()#Ligação à base de dados

        coluna=["Title","ArtistId"]#Lista com as colunas de albums

        #escolher a coluna de modificaçao
        print(coluna)
        vertical=input("Selecione a coluna onde quer modificar: ")#Pedir informação
        #ciclo que garante que é escolhida uma coluna válida
        while vertical not in coluna:
             vertical=input("Selecione a coluna onde quer modificar: ")

        #Opção Title
        if vertical=="Title":
            #Seleção dos valores o id e o nome do album
            cursor=Base.connect()
            cursor.execute("SELECT AlbumId,Title FROM albums")
            dados=cursor.fetchall()
            Base.connection.commit()
            print(" ")
            print('         -Albums-         ')

            #Selecionar os valores de id dos albums
            cursor=Base.connect()
            cursor.execute("SELECT AlbumId FROM albums")
            dados=cursor.fetchall()
            idr=[]
            Base.connection.commit()

            #Adicionar na lista todos os resultados
            for linha in range(len(dados)):
                idr.append(dados[linha][0])

            #Ciclo que pede e verifica se é um id válido
            while True:
                try:
                    print(" ")
                    id_album=input('Id do Album a alterar? ')
                    id_album=int(id_album)
                    if id_album in idr:
                        break
                    else:
                        print('ID não encontrado, tente novamente ')
                        print(" ")
                        continue
                except ValueError:
                    print('ID não encontrado, tente novamente')
                    print(" ")
                    op=input("Id do Album a alterar ")
            print(" ")
            album_novo=input(("Nome do album novo? "))

            #Atualizar para os dados novos
            cursor.execute("UPDATE albums SET Title= '%s' WHERE AlbumId = %i"%(album_novo,int(id_album)) )
            Base.connection.commit()
            print(" ")
            print("Alterado com sucesso ")

        #Opção ArtistId
        elif vertical=="ArtistId":
            #Seleção dos valores de albums
            cursor=Base.connect()
            cursor.execute("SELECT AlbumId,Title,ArtistId FROM albums")
            dados=cursor.fetchall()
            Base.connection.commit()

            print('            -Albums-         ')

            #Selecionar id dos albums
            cursor=Base.connect()
            cursor.execute("SELECT AlbumId FROM albums")
            dados=cursor.fetchall()
            ids_a=[]
            Base.connection.commit()

            #Adicionar na lista todos os resultados
            for linha in range(len(dados)):
                ids_a.append(dados[linha][0])

            #Ciclo que pede e verifica se é um id válido
            while True:
                try:
                    print(" ")
                    id_album=input('Qual o AlbumId que pretende utilizar para em seguida alterar o ArtistId? ')
                    id_album=int(id_album)
                    if id_album in ids_a:
                        break
                    else:
                        print(" ")
                        print('Nao encontrado, tente novamente! ')
                        continue
                except ValueError:
                    print(" ")
                    print('Nao encontrado, tente novamente! ')
                    op=input("Id a mudar: ")

            #Seleção dos valores dos artistas
            cursor=Base.connect()
            cursor.execute("SELECT ArtistId,Name FROM artists")
            dados=cursor.fetchall()
            l_artista=[]
            Base.connection.commit()

            #Adicionar na lista todos os resultados
            for linha in range(len(dados)):
                l_artista.append(dados[linha][0])

            #Ciclo que garante que existe o id e que é um número inteiro e válido
            while True:
                try:
                    print(" ")
                    id_a_novo=input("Novo ID do Artista? ")
                    id_a_novo=int(id_a_novo)
                    if id_a_novo in l_artista:
                        break
                    else:
                        print(" ")
                        print('Não encontrado, tente novamente ')
                        continue
                except ValueError:
                    print(" ")
                    print('Não encontrado, tente novamente')
                    print(" ")
                    id_a_novo=input("Id do artista: ")

            #Atualizar/substituir os dados antigos pelos novos
            cursor.execute("UPDATE albums SET ArtistId = '%s' WHERE AlbumId = '%s'"%(id_a_novo,int(id_album)))
            Base.connection.commit()
            print(" ")
            print("Alterado com sucesso ")
            print(" ")