Commit 74017bf2 authored by tarun2512's avatar tarun2512

second commit

parent 7088b4d6
...@@ -5,6 +5,5 @@ USER_ID=fpWx7Aw4NJzzXWxcE6w5dU ...@@ -5,6 +5,5 @@ USER_ID=fpWx7Aw4NJzzXWxcE6w5dU
PROJECT_ID=project_287 PROJECT_ID=project_287
APP_NAME=FT Energy Manager APP_NAME=FT Energy Manager
PREFIX=project_287 PREFIX=project_287
ZIP_NAME=FT Energy Manager.zip
BASE_PATH=/code/data BASE_PATH=/code/data
MOUNT_DIR=app_zip MOUNT_DIR=app_zip
\ No newline at end of file
import logging import logging
from pymongo import MongoClient from pymongo import MongoClient
import time
import shutil
from dotenv import load_dotenv from dotenv import load_dotenv
import os import os
import json
load_dotenv() load_dotenv()
import sys import sys
import zipfile
import httpx import httpx
from app_clone_script_helper import CloneScriptHelper from app_clone_script_helper import CloneScriptHelper
...@@ -22,7 +18,6 @@ USER_ID = os.environ.get("USER_ID") ...@@ -22,7 +18,6 @@ USER_ID = os.environ.get("USER_ID")
PROJECT_ID = os.environ.get("PROJECT_ID") PROJECT_ID = os.environ.get("PROJECT_ID")
APP_NAME = os.environ.get("APP_NAME") APP_NAME = os.environ.get("APP_NAME")
PREFIX = os.environ.get("PREFIX") PREFIX = os.environ.get("PREFIX")
ZIP_NAME = os.environ.get("ZIP_NAME")
BASE_PATH = os.environ.get("BASE_PATH") BASE_PATH = os.environ.get("BASE_PATH")
MOUNT_DIR = os.environ.get("MOUNT_DIR") MOUNT_DIR = os.environ.get("MOUNT_DIR")
...@@ -215,12 +210,12 @@ def push_app(folder_path, folder_name, user_id, project_id, prefix=None): ...@@ -215,12 +210,12 @@ def push_app(folder_path, folder_name, user_id, project_id, prefix=None):
connection.customer_apps.update_one(query, {"$set": data}, upsert=True) connection.customer_apps.update_one(query, {"$set": data}, upsert=True)
def push_data(app_name, folder_name, user_id, project_id, prefix=None): def push_data(zip_path, folder_name, user_id, project_id, prefix=None):
folder_split_data = folder_name.split(".") folder_split_data = folder_name.split(".")
connection = Connection() connection = Connection()
user_details = connection.user_conn.find_one({"user_id": user_id}, {"_id": 0}) user_details = connection.user_conn.find_one({"user_id": user_id}, {"_id": 0})
collection_name = folder_split_data[1] collection_name = folder_split_data[1]
folder_path = os.path.join(app_name, folder_name) folder_path = os.path.join(zip_path, folder_name)
if collection_name == "category": if collection_name == "category":
push_category(folder_path, folder_name, project_id, prefix) push_category(folder_path, folder_name, project_id, prefix)
elif collection_name == "widget": elif collection_name == "widget":
...@@ -235,14 +230,14 @@ def push_data(app_name, folder_name, user_id, project_id, prefix=None): ...@@ -235,14 +230,14 @@ def push_data(app_name, folder_name, user_id, project_id, prefix=None):
push_lookups(folder_path, folder_name, prefix) push_lookups(folder_path, folder_name, prefix)
def extract_app_zip(app_zip_name, app_name, user_id, project_id, prefix=None): def extract_app_zip(app_name, user_id, project_id, prefix=None):
zip_path = os.path.join(BASE_PATH, MOUNT_DIR, f"{app_name}.zip") zip_path = os.path.join(BASE_PATH, MOUNT_DIR, f"{app_name}.zip")
CloneScriptHelper().unzip_file(zip_path, zip_path) CloneScriptHelper().unzip_file(zip_path, zip_path)
folders_list = CloneScriptHelper().list_folders(app_name) folders_list = CloneScriptHelper().list_folders(zip_path)
for folder in folders_list: for folder in folders_list:
if folder in ["app_image", "app_logo"]: if folder in ["app_image", "app_logo"]:
continue continue
push_data(app_name, folder, user_id, project_id, prefix) push_data(zip_path, folder, user_id, project_id, prefix)
if __name__ == "__main__": if __name__ == "__main__":
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment