X-Git-Url: https://gerrit.akraino.org/r/gitweb?a=blobdiff_plain;f=scripts%2Fjcopy3.py;fp=scripts%2Fjcopy3.py;h=4ce92499eed1f5d4aabe3b1265e2e8110080fdd3;hb=8ca1343f22312d9711b92fed95ad52655842451a;hp=0000000000000000000000000000000000000000;hpb=674e68bc39ef4920273767fc5feb98a448fadab8;p=yaml_builds.git

diff --git a/scripts/jcopy3.py b/scripts/jcopy3.py
new file mode 100755
index 0000000..4ce9249
--- /dev/null
+++ b/scripts/jcopy3.py
@@ -0,0 +1,79 @@
+#!/usr/local/bin/python
+##############################################################################
+# Copyright © 2018 AT&T Intellectual Property. 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.                                             #
+##############################################################################
+
+#
+#  jcopy.py - Copy a file or files to a target directory, making
+#    substitutions as needed from the values contained in a YAML file.
+#
+#  usage: jcopy.py <yaml> <in_dir_or_file> <out_dir>
+#
+#  Note: jcopy.sh is for Python2, jcopy3.sh is for Python3
+#
+
+import os.path
+import jinja2
+import sys
+import yaml
+
+def expand_files(target_dir, dir_name, files):
+	global total
+	xlen = len(sys.argv[2])
+	targdir = target_dir + dir_name[xlen:]
+	if not os.path.exists(targdir):
+		os.makedirs(targdir)
+	env = jinja2.Environment()
+	for f in files:
+		source_path = dir_name + '/' + f
+		target_path = targdir + '/' + f
+		if os.path.isfile(source_path):
+			with open(source_path) as fd:
+				template = env.from_string(fd.read())
+			data = template.render(yaml=yaml)
+			fd2 = open(target_path,'w')
+			fd2.write(data)
+			fd2.close()
+			total += 1
+
+def expand_file(target_dir, file):
+	global total
+	if not os.path.exists(target_dir):
+		os.makedirs(target_dir)
+	env = jinja2.Environment()
+	with open(file) as fd:
+		template = env.from_string(fd.read())
+	data = template.render(yaml=yaml)
+	target_path = target_dir + '/' + os.path.basename(file)
+	fd2 = open(target_path,'w')
+	fd2.write(data)
+	fd2.close()
+	total += 1
+
+if len(sys.argv) != 4:
+	print('usage: jcopy.py <yaml> <in_dir_or_file> <out_dir>')
+	sys.exit(1)
+
+with open(sys.argv[1]) as f:
+	yaml = yaml.safe_load(f)
+
+total = 0
+if os.path.isfile(sys.argv[2]):
+	expand_file(sys.argv[3], os.path.abspath(sys.argv[2]))
+else:
+	for root, dirs, files in os.walk(sys.argv[2]):
+		expand_files(sys.argv[3], root, files)
+print('%d files processed.' % total)
+sys.exit(0)