pull/1000/merge
Jeongseop Lim 10 months ago committed by GitHub
commit fbf940f60f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -58,6 +58,19 @@ def C4Node(name, technology="", description="", type="Container", **kwargs):
return Node(**node_attributes) return Node(**node_attributes)
def Component(name, technology="", description="", **kwargs):
container_attributes = {
"name": name,
"technology": technology,
"description": description,
"type": "Component",
"fillcolor": "lightskyblue",
"fontcolor": "black",
}
container_attributes.update(kwargs)
return C4Node(**container_attributes)
def Container(name, technology="", description="", **kwargs): def Container(name, technology="", description="", **kwargs):
container_attributes = { container_attributes = {
"name": name, "name": name,

@ -5,7 +5,7 @@ import unittest
from diagrams import Diagram from diagrams import Diagram
from diagrams import setcluster, setdiagram from diagrams import setcluster, setdiagram
from diagrams.c4 import Person, Container, Database, System, SystemBoundary, Relationship from diagrams.c4 import Person, Container, Database, Component, System, SystemBoundary, Relationship
class C4Test(unittest.TestCase): class C4Test(unittest.TestCase):
@ -25,6 +25,7 @@ class C4Test(unittest.TestCase):
person = Person("person", "A person.") person = Person("person", "A person.")
container = Container("container", "Java application", "The application.") container = Container("container", "Java application", "The application.")
database = Database("database", "Oracle database", "Stores information.") database = Database("database", "Oracle database", "Stores information.")
component = Component("component", "Sign In Controller", "A component.")
def test_external_nodes(self): def test_external_nodes(self):
with Diagram(name=self.name, show=False): with Diagram(name=self.name, show=False):

Loading…
Cancel
Save