@acwoss/

RawDelightfulAmericanalligator

Python

No description

fork
loading
Files
  • main.py
  • filmes.csv
main.py
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
import csv

def search_by_genre(genre, filename='filmes.csv'):
    """ Busca em um arquivo todos os filmes de um determinado gênero.
    
    Atributos:
        genre (string): Nome do gênero a ser buscado.
        filename (string): Nome do arquivo onde os filmes estão armazenados.
        
    Retorno:
        Retorna um gerador (iterável) com todos os filmes referentes ao gênero.
        
    Exceções:
        FileNotFoundError: Quando o arquivo indicado por `filename` não existir.
    """
    with open(filename, 'r') as stream:
        reader = csv.DictReader(stream, delimiter=';')
        for row in reader:
            if row['genre'].lower() == genre.lower():
                yield row
    
if __name__ == '__main__':
    genre = input('Digite o gênero do filme: ')
    movies = search_by_genre(genre, filename='filmes.csv')
    for movie in movies:
        print(movie['name'])