Package: erdgen
erdgenDBT YML ERD GeneratorOverviewThis Python program generates Database Markup Language (DBML) Entity Relationship Diagram's (ERD) from the relationships node in your dbt YML files. The script parses the YML files, extracts relationships and columns, and outputs a DBML schema.The program is pretty opinionated. It requires each YML file to only contain one model. Further, the "relationships" node in dbt yml is a made-up construct.This program is useful for automated ERD generation if your dbt project doesn't have referential integrity or explicit SQL relationships. If your SQL models have defined SQL relationships there are better tools for automated ERD generation.Usagepython-merdgen--directory<directory>--include_non_join_keys<True/False>Args--directory: Directory to search for YAML files. The default value is the current directory ('.').--include_non_join_keys: Boolean flag to indicate whether to include non-join keys in the DBML. The default value isFalse.The DBML will be printed to the console. You can redirect this output to a file if desired.YML File StructureThe YML files are expected to have the following structure:version: 2

models:
    - name: Computer
      description: beep boop beep
      columns:
          - name: computerId
            description: The unique identifier of computer
          # other non-join key columns as necessary
      relationships:
          - name: files
            description: The files are in the computer!?
            type: one_to_many
            table: computer_files
            join:
                - local: computerId
                  remote: computerIdnote: Each YML file should contain only one model under themodelsnode.RelationshipsTherelationshipsnode in the YML files represents the relationship between the current model and other models. It is composed of several sub-nodes:name: The name of the relationship.description: A brief description of the relationship.type: The type of the relationship. It can beone_to_one,one_to_many,many_to_one, ormany_to_many.table: The name of the other model involved in the relationship.join: A list of the columns that are used for the join between the current model and the other model. Each item in the list is composed oflocaland -remotenodes, representing the column in the current model and the column in the other model, respectively.OutputThe output is a DBML schema that includes the tables, columns, and references based on the relationships defined in the YML files. The output is printed to the console.NotesIf a YML model file has norelationships, andinclude_non_join_keysisFalse, all columns from the YML are included in the DBML table. This is helpful as other models may have arelationshipto this model, and there is no way to know which column is being referenced (well there is but I didn't bother figuring this out)Regardless of whetherinclude_non_join_keysisTrueorFalse, columns that containIdoridin them are always included. These are likely join keys that do not have a relationship yet.ImprovementsAll data types are int, account for the actual data type via metadata in the YML fileAll relationships are 1:1, account for the cardinality via the relationshiptypeWhat about.yamlfiles lolDEVCreate venvpython-mvenvenvActivate venvunixsourceenv/bin/activatewindowsenv\Scripts\activate.batInstall Packagespipinstall-rrequirements.txtTestmaketestFormatmakeformatmakelintVersion & Releasebump2version<major/minor/patch>makereleasenoteDon't forget togit pushwith--tagspre-commitSetuppre-commitinstallRun allmakepre-commit