3dtiles_mongo write3dtilesModelToMongo.py

最新推荐文章于 2024-10-10 08:38:55 发布

学无止境慢慢来

最新推荐文章于 2024-10-10 08:38:55 发布

阅读量854

点赞数

CC 4.0 BY-SA版权

分类专栏： 3dtiles 模型分析-建库-反向查询

本文链接：https://blog.youkuaiyun.com/weixin_42193179/article/details/94744806

"""
此模块用于，把原始BIM模型“小学-机电合模”的文件写到本地mongodb的BIMModel数据库。
"""
import os
from pymongo import MongoClient
# 以下两行代码，放到if __name__ == "__main__":下会报错。
# 不知原因。
client = MongoClient()
# 确定数据库。
db = client.BIMModel


def getPathFilesInDirectory(directory, pathFiles=[]):
    """
    此函数用于，获取directory目录下，所有文件的绝对路径，
    并存放在pathFiles里。
    directory下可以同时有目录和文件。 
    如果在调用此函数时，未对pathFiles传参，
    则pathFiles列表只在函数定义时创建，分配固定起始地址的存储空间。
    """
    filesFolders = os.listdir(directory)
    # print(filesFolders)
    for f in filesFolders:
        pathFileOrFolder = os.path.join(directory, f)
        if os.path.isfile(pathFileOrFolder):
            # print(pathFileOrFolder)
            pathFiles.append(pathFileOrFolder)
        else:
            getPathFilesInDirectory(pathFileOrFolder, pathFiles)
    return pathFiles


def writeFileToMongodb(pathFile):
    """
    此函数用于，写一个文件到mongodb的BIM数据库。
    集合名，小学-机电合模。
    :param pathFile:
    :return:
    """
    pri