get_local_models.py

#!/usr/bin/env python3
from __future__ import print_function

import sys
import os
import shutil
import distutils.core


def copy_models(src, dst):

    list_dir = [os.path.join(src, x) for x in os.listdir(src)]
    sub_dirs = [x for x in list_dir if os.path.isdir(x)]
    for model_path in sub_dirs:
        if 'model.config' in os.listdir(model_path):
            # sys.stdout.write(" copying %s" % model_path)
            print(".", end="")
            dest_dir = os.path.join(dst, os.path.split(model_path)[1])
            if os.path.isdir(dest_dir):
                shutil.rmtree(dest_dir)
                print("overriding model %s" % model_path)
            shutil.copytree(model_path, dest_dir)
        else:
            print(" %s ignored" % model_path)


dest_dir = sys.argv[1]


print("copying local models to %s" % dest_dir)

unique_paths = None
try:
    gazebo_path = os.environ['GAZEBO_MODEL_PATH'].split(':')

    model_paths = [x for x in gazebo_path if os.path.isdir(x)]
    unique_paths = list(set(model_paths))
except:
    print("No local models.")
#    exit(0)

if unique_paths is not None:
    for path in unique_paths:
        print("\nmodel path: [%s]" % path)
        copy_models(path, dest_dir)

    print("local models transfered")


print("copying local resources to %s" % dest_dir)


def copy_resources(src, dst):
    media_path = os.path.join(src, "media")
    if os.path.exists(media_path) and os.path.isdir(media_path):
        dest_dir = os.path.join(dst, "media")
        distutils.dir_util.copy_tree(media_path, dest_dir)


unique_paths = None
try:
    resource_path = os.environ['GAZEBO_RESOURCE_PATH'].split(':')

    resource_paths = [x for x in resource_path if os.path.isdir(x)]
    unique_paths = list(set(resource_paths))
except:
    print("No local resources")
    exit(0)

if unique_paths is not None:
    for path in unique_paths:
        print("\nresource path: [%s]" % path)
        copy_resources(path, dest_dir)

print("local resources transfered")