예제 #1
0
파일: models.py 프로젝트: rajraj889/Auto-DL
 def __init__(self, username, password, **kwargs):
     self.username = username
     self.password = password
     self.db = connector.connect()
     self.collection_name = kwargs.get('collection', 'user')
     self.collection = self.db[self.collection_name]
     self.attributes = kwargs
예제 #2
0
 def __init__(self, username, password, **kwargs):
     self.username = username
     self.password = password
     self.is_verified = False
     self.db = connector.connect()
     self.collection_name = kwargs.get("collection", "user")
     self.collection = self.db[self.collection_name]
     self.attributes = kwargs
예제 #3
0
파일: models.py 프로젝트: rajraj889/Auto-DL
 def __init__(self, user, **kwargs):
     self.user = user
     self.db = connector.connect()
     self.collection_name = kwargs.get('collection', 'session')
     self.collection = self.db[self.collection_name]
     self.attributes = kwargs
예제 #4
0
import sys

sys.path.append("..")

from django.db import models
from authv1.connector import connect

db = connect(db_name="user_data")


class UserData:
    def __init__(self, meta, components, layers, parsed_layers, preprocess):
        self.meta = meta
        self.components = components
        self.layers = layers
        self.preprocess = preprocess
        self.parsed_layers = parsed_layers
        self.collection = db["ir"]

    def upload(self):
        data_doc = {
            "meta": self.meta,
            "components": self.components,
            "layers": self.layers,
            "preprocess": self.preprocess,
            "parsed_layers": self.parsed_layers,
        }
        try:
            self.collection.insert_one(data_doc)
            return {"status": 1, "msg": "success"}
        except Exception as e: