From ab82d77d8ec680de5a59952f7fb2617df454faa2 Mon Sep 17 00:00:00 2001
From: wangye <wangye@i-i.ai>
Date: Sat, 24 Aug 2024 14:56:09 +0800
Subject: [PATCH] add arg yaml_path

---
 main.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index c525244..c286695 100644
--- a/main.py
+++ b/main.py
@@ -6,6 +6,7 @@
 import os
 import re
 import shutil
+import sys
 from pathlib import Path
 from yaml import load, Loader
 
@@ -110,7 +111,7 @@ class Watcher:
     startup: bool
 
     def __init__(self):
-        stream = io.open("config.yaml", "r")
+        stream = io.open(yaml_path, "r")
         config = load(stream, Loader)
         self.name = config.get("name", "no name")
         self.startup = config.get("startup", False)
@@ -160,4 +161,8 @@ def start(self):
 
 
 if __name__ == '__main__':
+    if len(sys.argv) == 1:
+        yaml_path = "config.yaml"
+    else:
+        yaml_path = sys.argv[1]
     Watcher().start()