Skip to content

Commit

Permalink
Adiciona a model
Browse files Browse the repository at this point in the history
  • Loading branch information
benlacerda committed Aug 21, 2024
1 parent 47f2d6e commit 5ea2143
Showing 1 changed file with 54 additions and 0 deletions.
54 changes: 54 additions & 0 deletions src/recommendation_model/utils/model.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,54 @@
from typing import List, Optional

class IVideo:
def __init__(self, id: int, title: str, description: Optional[str] = ''):
self.id = id
self.title = title
self.description = description
self.catalog = None # Será preenchido após a categorização

class Catalog:
def __init__(self):
self.journalism = {
"falaJovem": [],
"informeUnB": [],
"zapping": []
}
self.interviews = {
"brasilEmQuestao": [],
"dialogos": [],
"tirandoDeLetra": [],
"entrevistas": [],
"vastoMundo": [],
"vozesDiplomaticas": []
}
self.researchAndScience = {
"expliqueSuaTese": [],
"fazendoCiencia": [],
"radarDaExtencao": [],
"seLigaNoPAS": [],
"unbTvCiencia": [],
"universidadeParaQue": []
}
self.artAndCulture = {
"emCantos": [],
"casaDoSom": [],
"esbocos": [],
"exclusiva": []
}
self.specialSeries = {
"florestaDeGente": [],
"guiaDoCalouro": [],
"memoriasPauloFreire": [],
"desafiosDasEleicoes": [],
"vidaDeEstudante": [],
"arquiteturaICC": []
}
self.documentaries = {
"miniDoc": [],
"documentaries": []
}
self.varieties = {
"pitadasDoCerrado": []
}
self.unbtv = []

0 comments on commit 5ea2143

Please sign in to comment.