From f58ea58bd691ec39728bfeaf32654bfda42e24f9 Mon Sep 17 00:00:00 2001
From: leafee98 <leafee98@hotmail.com>
Date: Wed, 26 Mar 2025 00:30:07 +0800
Subject: [PATCH] Add check for existence of src and dst file

---
 copy-cert.py | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

diff --git a/copy-cert.py b/copy-cert.py
index 2477309..e1ee1bd 100644
--- a/copy-cert.py
+++ b/copy-cert.py
@@ -19,6 +19,9 @@ http_pool = urllib3.PoolManager()
 class InvalidConfigException(Exception):
     pass
 
+class FileNotExistsException(Exception):
+    pass
+
 
 class CopyEntry:
     def __init__(self, d: Dict):
@@ -46,15 +49,21 @@ class Config:
             self.certs.append(CertEntry(cert_d))
 
 def copyFile(src: str, dst: str):
-    src_mtime = os.path.getmtime(src)
-    dst_mtime = os.path.getmtime(dst)
+    if not os.path.exists(src):
+        raise FileNotExistsException('Src file not exists: {}'.format(src))
 
-    if src_mtime > dst_mtime:
+    if not os.path.exists(os.path.dirname(dst)):
+        raise FileNotExistsException('Direcotyr of dst not exists: {}'.format(dst))
+
+    is_dst_exists = os.path.exists(dst)
+    src_mtime = os.path.getmtime(src)
+    dst_mtime = os.path.getmtime(dst) if is_dst_exists else 0
+
+    if is_dst_exists and src_mtime < dst_mtime:
+        logger.info('Skipping file: {} --> {}'.format(src, dst))
+    else:
         logger.info('Copying file: {} --> {}'.format(src, dst))
         shutil.copyfile(src, dst)
-    else:
-        logger.info('Skipping file: {} --> {}'.format(src, dst))
-
 
 def copyCert(entry: CertEntry):
     copyFile(entry.priv.src, entry.priv.dst)