mirror of
https://github.com/Hestia-Homes/survey-extraction.git
synced 2026-06-30 13:10:56 +00:00
50 lines
2.1 KiB
Python
50 lines
2.1 KiB
Python
import os
|
|
os.environ["SHAREPOINT_CLIENT_ID"] = "0e28c4f9-3e77-4571-8d63-df1857f4266a"
|
|
os.environ["SHAREPOINT_CLIENT_SECRET"] = "2s48Q~t8.pI-~rbtQaSCFcYY97Z3LiMYhuo0GaOb"
|
|
os.environ["SHAREPOINT_TENANT_ID"] = "6f080c63-8a66-4bbc-9d72-b85d5df30555"
|
|
|
|
from etl.scraper.scraper import SharePointScraper, SharePointInstaller
|
|
from datetime import datetime
|
|
|
|
|
|
def upload_to_month_end_folder(file_name_on_sp, local_file_path, add_to_path):
|
|
sharepoint = SharePointScraper(SharePointInstaller.OSMOSIS_ACD)
|
|
sharepoint.get_folders_in_path("/")
|
|
|
|
parent_folder = "General/Junte Kim/month end"
|
|
today = datetime.today()
|
|
# Format as "Month YYYY"
|
|
formatted_date = today.strftime("%B %Y")
|
|
sharepoint.create_dir(formatted_date, parent_folder)
|
|
sharepoint_path = parent_folder + "/" + formatted_date
|
|
|
|
|
|
# Make day month year folder
|
|
formatted_date = today.strftime("%d-%m-%y")
|
|
sharepoint.create_dir(formatted_date, sharepoint_path)
|
|
sharepoint_path += "/" + formatted_date
|
|
|
|
# Make company folder
|
|
sharepoint.create_dir(add_to_path, sharepoint_path)
|
|
sharepoint_path += "/" + add_to_path
|
|
|
|
print("Uploading to sharepoint...")
|
|
sharepoint.upload_file(local_file_path, sharepoint_path, file_name_on_sp)
|
|
print(f"Finished upload of {local_file_path} to sharepoint. It's found under {sharepoint_path}/{file_name_on_sp}")
|
|
|
|
|
|
def upload_to_nick_folder(file_name_on_sp, local_file_path, add_to_path=None):
|
|
sharepoint = SharePointScraper(SharePointInstaller.OSMOSIS_ACD)
|
|
parent_folder = "General/Junte Kim/For Nick"
|
|
today = datetime.today()
|
|
formatted_date = today.strftime("%Y-%m-%d%H-%M-S")
|
|
|
|
sharepoint.create_dir(formatted_date, parent_folder)
|
|
sharepoint_path = parent_folder + "/" + formatted_date
|
|
if add_to_path:
|
|
sharepoint.create_dir(add_to_path, sharepoint_path)
|
|
sharepoint_path += "/" + add_to_path
|
|
|
|
print("Uploading to sharepoint...")
|
|
sharepoint.upload_file(local_file_path, sharepoint_path, file_name_on_sp)
|
|
print(f"Finished upload of {local_file_path} to sharepoint. It's found under {sharepoint_path}/{file_name_on_sp}")
|