Beispiel #1
0
    def __init__(self, *args, **kwargs):

        if len(args) == 1:
            if isinstance(args[0], str):
                #        systemInstance=args[0]
                maxQueueSize = 10
            if isinstance(args[0], int):
                maxQueueSize = args[0]
#        systemInstance='Default'
        elif len(args) == 2:
            #      systemInstance=args[0]
            maxQueueSize = args[1]
        elif len(args) == 0:
            #      systemInstance='Default'
            maxQueueSize = 10

        if 'DBin' in kwargs.keys():
            DBin = kwargs['DBin']
            if isinstance(DBin, list):
                from DIRAC.Core.Utilities.MySQL import MySQL
                self.db = MySQL('localhost', DBin[0], DBin[1],
                                'ResourceManagementDB')
            else:
                self.db = DBin
        else:
            from DIRAC.Core.Base.DB import DB
            self.db = DB('ResourceManagementDB',
                         'ResourceStatus/ResourceManagementDB', maxQueueSize)
Beispiel #2
0
    def __init__(self, *args, **kwargs):
        """Constructor."""
        if len(args) == 1:
            if isinstance(args[0], str):
                maxQueueSize = 10
            if isinstance(args[0], int):
                maxQueueSize = args[0]
        elif len(args) == 2:
            maxQueueSize = args[1]
        elif len(args) == 0:
            maxQueueSize = 10

        if 'DBin' in kwargs.keys():
            dbIn = kwargs['DBin']
            if isinstance(dbIn, list):
                from DIRAC.Core.Utilities.MySQL import MySQL
                self.db = MySQL('localhost', dbIn[0], dbIn[1],
                                'ResourceManagementDB')
            else:
                self.db = dbIn
        else:
            from DIRAC.Core.Base.DB import DB
            self.db = DB('ResourceManagementDB',
                         'ResourceStatus/ResourceManagementDB', maxQueueSize)

        self.mm = MySQLMonkey(self)
    def __init__(self, mySQL=None):
        '''
      Constructor, accepts any DB or mySQL connection, mostly used for testing
      purposes.
    '''
        self._tableDict = self.__generateTables()

        if mySQL is not None:
            self.database = mySQL
        else:
            self.database = DB('ResourceManagementDB',
                               'ResourceStatus/ResourceManagementDB')