Construirea unui pipeline de date

Utilizarea metodelor factory nu doar că face codul mai ușor de citit, ci și mai ușor de folosit. În acest exemplu, vei exersa crearea unui pipeline de date care extrage date dintr-o bază de date. Clasa DataPipeline implementează pattern-ul de design factory method și este prezentată mai jos. De asemenea, sunt definite pentru tine două produse concrete ale clasei Database: Postgres și Redshift.

class DataPipeline:
  def _get_database(self, provider):
    if provider == "Postgres":
      return Postgres()
    elif provider == "Redshift":
      return Redshift()

  def extract_data(self, provider, query):
    database = self._get_database(provider)
    dataset = database.query_data(query)
    print(f"Extracted dataset from {provider} database")
    return dataset