示例#1
0
文件: db_lib.py 项目: ericdill/lsdc
def getResultsforBL(id=None, name=None, number=None):
    """
    Retrieve results using either BL id, name, or number (tried in that order)
    Returns a generator of results
    """
    if id is None:
        if name is None:
            key = 'number'
            val = number
        else:
            key = 'name'
            val = name

        query = {key: val}
        b = Beamline.objects(__raw__=query)

        id = _try0_dict_key(b, 'beamline', key, val, None, 'beamline_id')

        if id is None:
            yield None
            raise StopIteration

    for r in Result.objects(__raw__={'beamline_id': id}):
        yield r
示例#2
0
文件: db_lib.py 项目: ericdill/lsdc
def getBeamlineByNumber(num, as_mongo_obj=False):
    """eg. 17id1, 17id2, 16id1"""
    b = Beamline.objects(__raw__={'number': num})
    return _try0_maybe_mongo(b, 'beamline', 'number', num, None,
                             as_mongo_obj=as_mongo_obj)
示例#3
0
文件: db_lib.py 项目: ericdill/lsdc
def getBeamlineByID(beamline_id, as_mongo_obj=False):
    """beamline_id is the integer field 'beamline_id'"""
    b = Beamline.objects(__raw__={'beamline_id': beamline_id})
    return _try0_maybe_mongo(b, 'beamline', 'beamline_id', beamline_id, None,
                             as_mongo_obj=as_mongo_obj)
示例#4
0
文件: db_lib.py 项目: ericdill/lsdc
def getBeamlineByName(name, as_mongo_obj=False):
    """eg. fmx, xfp, chx..."""
    b = Beamline.objects(__raw__={'name': name})
    return _try0_maybe_mongo(b, 'beamline', 'name', name, None,
                             as_mongo_obj=as_mongo_obj)
示例#5
0
文件: db_lib.py 项目: ericdill/lsdc
def createBeamline(bl_name, bl_num):
    bl = Beamline(name=bl_name, number=bl_num)
    bl.save()