check-up/create_db.py at develop · EH-FAKE/check-up · GitHub

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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
from decouple import config
from sqlalchemy import create_engine
from sqlalchemy.orm import Session

from models import (
    Advertisement,
    Portal,
    Entry,
    QueueStatus,
    URLQueue,
)


if __name__ == "__main__":
    engine = create_engine(config("DATABASE_URL"))

    # Primeiro, dropa todas as tabelas existentes
    Advertisement.metadata.drop_all(engine)
    URLQueue.metadata.drop_all(engine)
    Entry.metadata.drop_all(engine)
    Portal.metadata.drop_all(engine)
    QueueStatus.metadata.drop_all(engine)

    # Recria todas as tabelas
    Portal.metadata.create_all(engine)
    Entry.metadata.create_all(engine)
    Advertisement.metadata.create_all(engine)
    URLQueue.metadata.create_all(engine)
    QueueStatus.metadata.create_all(engine)

    portals = [
        ("UOL", "https://noticias.uol.com.br/", "uol"),
        ("Folha", "https://www.folha.uol.com.br/", "folha"),
        ("Estadão", "https://www.estadao.com.br/", "estadao"),
        ("Veja", "https://veja.abril.com.br/", "veja"),
        ("Terra", "https://www.terra.com.br/", "terra"),
        ("Metropoles", "https://www.metropoles.com/", "metropoles"),
        ("ClicRBS", "https://clicrbs.com.br/", "clicrbs"),
        ("Globo", "https://oglobo.globo.com/", "globo"),
        ("IG", "https://www.ig.com.br/", "ig"),
        ("R7", "https://www.r7.com/", "r7"),
        ("Gazeta do Povo", "https://www.gazetadopovo.com.br/", "gazetadopovo"),
        ("Mais Goiás", "https://www.maisgoias.com.br/", "maisgoias"),
        ("Aliados Brasil", "https://www.aliadosbrasiloficial.com.br/", "aliadosbrasil"),
    ]
    portals_to_add = []
    for portal in portals:
        portals_to_add.append(Portal(name=portal[0], url=portal[1], slug=portal[2]))

    with Session(engine) as session:
        session.add_all(portals_to_add)
        session.commit()