sihunh / yarateste

웹사이트 내 Yara Rule를 이용해 webshell을 탐지하는 툴입니다 및 사용자가 다운로드 받은 파일에 대해 검사를 진행합니다
0 stars 0 forks source link

firebase cloud db 연동 #6

Open sihunh opened 1 year ago

sihunh commented 1 year ago

import json import firebase_admin from firebase_admin import credentials from firebase_admin import firestore

def firebase_upload(json_dt, colletion_name, file_name): cred = credentials.Certificate('fire_base_key.json') firebase_admin.initialize_app(cred)

db = firestore.client()

class UploadJsonFileToFirestore:
    def __init__(self) -> None:

        #세팅값
        self.json_data = json_dt #파일 이름도 됨 
        self.collectionname = colletion_name # 초기 컬렉션

    @property
    def json_data(self):
        return self._json_data

    #여기는 나중에 지울부분 
    @json_data.setter
    def json_data(self, val):
        if val:
            try:
                f = open(val,)
                data = json.load(f)
                f.close()
                self._json_data = data
            except Exception as e:
                print(f'FILE EXCEPTION: {str(e)}')
        else:
            print(f'Wrong file path {val}')

    def upload(self):
        if  self.json_data:
            self.set()

    def set(self):
        return db.collection(self.collectionname).document('e').set(self.json_data)

uploadjson = UploadJsonFileToFirestore()
uploadjson.upload()
sihunh commented 1 year ago

수정

sihunh commented 1 year ago

import firebase_admin from firebase_admin import credentials from firebase_admin import firestore

def firebase_upload(json_data, collection_name, file_name): cred = credentials.Certificate('fire_base_key.json') firebase_admin.initialize_app(cred)

db = firestore.client()
db.collection(collection_name).document(file_name).set(json_data)