fix: accept g6 from args in drawer

1 files changed, 6 insertions, 3 deletions

diff --git a/drawer/src/drawer/__init__.py b/drawer/src/drawer/__init__.py
index ef47fbc..dbe688e 100644
--- a/drawer/src/drawer/__init__.py
+++ b/drawer/src/drawer/__init__.py
@@ -1,8 +1,12 @@
 import networkx as nx
 import tempfile
+import sys
 
 def main():
-    g6 = "I?opuIiBw"
+    if len(sys.argv) < 2:
+        print("Необходимо передать g6 строку графа в качестве первого параметра")
+        exit(1)
+    g6 = sys.argv[1]
     with tempfile.NamedTemporaryFile(delete=False) as f:
         _ = f.write(g6.encode())
         _ = f.seek(0)
@@ -20,5 +24,4 @@ def main():
         node_options=node_options,
         edge_options=edge_options
     ))
-    # nx.write_latex(g, "graph.tex", as_document=True)
-    # nx.draw(g)
+