From 0599655e9c88b2e8b3d2c7c13b56c9aa8fb9de58 Mon Sep 17 00:00:00 2001
From: Yangrl <2535184404@qq.com>
Date: Tue, 23 Jan 2024 08:32:42 +0000
Subject: [PATCH] add_script,test=document_fix

---
 paddle/scripts/get_target_size.py | 51 +++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 paddle/scripts/get_target_size.py

diff --git a/paddle/scripts/get_target_size.py b/paddle/scripts/get_target_size.py
new file mode 100644
index 0000000000000..ec0ebd34d7955
--- /dev/null
+++ b/paddle/scripts/get_target_size.py
@@ -0,0 +1,51 @@
+# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import subprocess
+import sys
+
+
+class HandleTarget:
+    def __init__(self) -> None:
+        argv = sys.argv
+        len_ = len(argv)
+        fromPath = './' if len_ <= 1 else argv[1]
+        cmd = f"find {fromPath} -name '*.o' -not -path './third_party/*' | xargs du -sch > log"
+        subprocess.run(cmd, shell=True)
+
+    def calcuSize(self, item):
+        size = float(item[:-1])
+        res = size * 1024 if item.find('G') >= 0 else size
+        return size / 1024 if item.find('K') >= 0 else res
+
+    def getSize(self):
+        ctx = self.getDatas()
+        sum = 0
+        for item in ctx:
+            if item.find('total') >= 0:
+                item = item.split('\t')[0]
+                sum += self.calcuSize(item)
+
+        print(f"Total size is {sum} M (without third_party)")
+
+    def getDatas(self):
+        with open('log', 'r') as file:
+            ctx = file.read()
+        ctx = ctx.split('\n')
+        return ctx
+
+
+if __name__ == '__main__':
+    handler = HandleTarget()
+    handler.getSize()