def recover_devices_recycles(ids): devices = Device.objects(Q(id__in=ids) & Q(is_template=False)) devices.update(is_deleted=False) equipments = [equipment.id for equipment in devices.values_list('_parent_equipment')] Equipment.objects(id__in=equipments).update(is_deleted=False) return jsonify(ok=True, msg='success')
def delete_all(): Sample.objects.all().delete() Collect.objects().all().delete() LoanHistory.objects().all().delete() Equipment.objects().all().delete() Sensor.objects().all().delete() Patient.objects().all().delete() return jsonify({'message': 'deleted with success.'})
async def create_equipment(equipment: Equipment, user: User = Depends(is_current_active_user_admin)): if hasattr(equipment, 'id'): delattr(equipment, 'id') _equipment = db.equipments.insert_one(equipment.dict(by_alias=True)) equipment.id = _equipment.inserted_id return equipment
def get(self): _id = request.args.get("id") equip = Equipment() try: equip.payload = hdb.get(equip, _id) return equip.payload except Exception as error: gunicorn_logger.error(error) return "Could not retrieve equipment", 400
def create_vessel(): """ """'' record = json.loads(request.data) name, code, location, status, vessel = (record['name'] if "name" in record.keys() else None), (record['code'] if "code" in record.keys() else None), (record['location'] if "location" in record.keys() else None),(record['status'] if "status" in record.keys() else None), (record['vessel'] if "vessel" in record.keys() else None) equipment = Equipment(name=name, code=code,location=location,status= vessel, vessel= vessel) equipment.save() return jsonify(equipment.to_json()), 201
def post(self): new_equipment = Equipment() valid = new_equipment.create(request.json) if valid: response = hdb.insert(new_equipment) return { "id": new_equipment._id, "status": response, "equipment": new_equipment.payload, } else: return "Could not create equipment", 400
def recover_interfaces_recycles(ids): interfaces = Interface.objects(Q(id__in=ids) & Q(is_template=False)) interfaces.update(is_deleted=False) # 恢复接口自身 devices = [ device.id for device in interfaces.values_list('_parent_device') ] equipments = [ equipment.id for equipment in interfaces.values_list('_parent_equipment') ] Device.objects(id__in=devices).update(is_deleted=False) Equipment.objects(id__in=equipments).update(is_deleted=False) return jsonify(ok=True, msg='success')
def create_equipment(): request_data = request.get_json() for sensor in request_data['sensors']: sensor['_id'] = ObjectId(sensor['$oid']) del sensor['$oid'] equipment = Equipment(**request_data).save() return jsonify(equipment)
def get_equipments(): form = SearchForm(request.args) query = Equipment.objects(is_deleted=False, is_template=False) # 简单搜索 if form.searchKey.data: search_key = form.searchKey.data query = query.filter(Q(name__contains=search_key) | Q(note__contains=search_key)) return build_order_by(query, form.orderBy.data), 200 # 高级搜索 if form.name.data: query = query.filter(name__contains=form.name.data) if form.type.data: query = query.filter(type=form.type.data) if form.location.data: query = query.filter(location=form.location.data) if form.sn.data: query = query.filter(sn=form.sn.data) if form.manufacturer.data: query = query.filter(manufacturer=form.manufacturer.data) if form.note.data: query = query.filter(note__contains=form.note.data) if form.createUserId.data: query = query.filter(_create_user=form.createUserId.data) if form.createdAtStart.data: query = query.filter(created_at__gte=form.createdAtStart.data) if form.createdAtEnd.data: query = query.filter(created_at__lte=form.createdAtEnd.data) current_user.update(inc__search_num=1) return build_order_by(query, form.orderBy.data)
def get_users_meta(): data = Equipment.objects(is_deleted=False, is_template=False).fields(id=1, name=1).all() meta = [{ 'id': str(item.id), 'name': item.name, } for item in data] return jsonify(ok=True, data=meta)
def get(self): equip_type = request.args.get("equip_type") try: response = hdb.get_list(Equipment(), equip_type, "equip_type") return {"equipment": response} except Exception as error: gunicorn_logger.error(error) return "Could not retrieve equipment", 400
def get_child(oid): equipment = Equipment.objects(id=oid).count() if not equipment: return jsonify(ok=True, msg='not found'), 404 child_device = Device.objects(_parent_equipment=oid).count() child_interface = Interface.objects(_parent_equipment=oid).count() child_cable = Cable.objects(Q(_start_equipment=oid) | Q(_end_equipment=oid)).count() return jsonify(ok=True, msg={'device': child_device, 'interface': child_interface, 'cable': child_cable})
def index(vessel): equipments = Equipment.objects(vessel=vessel) print(equipments) response = [] for equipment in equipments: response.append(equipment.to_json()) if not equipments: return jsonify({'error': 'data not found'}) else: return jsonify(response)
def get_equipments_child(): form = ObjectIdListForm(request.args) ids = form.ids.data equipment = Equipment.objects(id__in=ids).count() if not equipment: return jsonify(ok=True, msg='not found'), 404 child_device = Device.objects(_parent_equipment__in=ids).count() child_interface = Interface.objects(_parent_equipment__in=ids).count() child_cable = Cable.objects(Q(_start_equipment__in=ids) | Q(_end_equipment__in=ids)).count() return jsonify(ok=True, msg={'device': child_device, 'interface': child_interface, 'cable': child_cable})
def deleteVessel(code): equipment = Equipment.objects(code=code).first() if not equipment: return jsonify({'error': 'data not found'}), 404 else: try: equipment.delete() return jsonify(), 204 except Exception as e: print(e) return jsonify(), 204
def deactivateEquipment(): listOfEquipments = json.loads(request.data) if not listOfEquipments: return jsonify({'error': 'data not found'}), 404 for equipment in listOfEquipments: equipments_doc = Equipment.objects(code=equipment).first() if not equipments_doc: return jsonify({'error': 'data not found'}), 404 else: equipments_doc.update(status=False) return jsonify(equipments_doc.to_json())
def recover_cables_recycles(ids): cables = Cable.objects(Q(id__in=ids) & Q(is_template=False)) cables.update(is_deleted=False) # 恢复线缆自身 interfaces = [ interface.id for interface in cables.values_list( '_start_interface', '_end_interface')[0] ] devices = [ device.id for device in cables.values_list('_start_device', '_end_interface')[0] ] equipments = [ equipment.id for equipment in cables.values_list( '_start_equipment', '_end_interface')[0] ] Interface.objects(id__in=interfaces).update(is_deleted=False) Device.objects(id__in=devices).update(is_deleted=False) Equipment.objects(id__in=equipments).update(is_deleted=False) return jsonify(ok=True, msg='success')
def load(self, device): device.name = self.name.data device.type = self.type.data device.location = self.location.data device.system = self.system.data device.sn = self.sn.data device.manufacturer = self.manufacturer.data device.note = self.note.data try: parent_equipment = Equipment.objects().get(id=self.parentEquipmentId.data) except Equipment.DoesNotExist: raise FormLoadException('装置不存在,id: ' + str(self.parentEquipmentId.data)) else: device.parent_equipment = parent_equipment.id
def recover_equipments_recycles(ids): Equipment.objects(id__in=ids).update(is_deleted=False) return jsonify(ok=True, msg='success')
from models.camera import Camera from models.equipment import Equipment from models.event import Event from models.floor import Floor from models.person import Person from models.room import Room from models.section import Section from models.typeofequipments import Type_Of_Equipments type1 = Type_Of_Equipments("Tables") print(type1) print("\n") q1 = Equipment(34, type1) print(q1) print("\n") f1 = Floor(1) p1 = Person("Иванов Иван Иванович", 1234) print(p1) print("\n") r1 = Room(101) r1.add_equipment(q1) f1.add_room_on_floor(r1) print(f1) print("\n") s1 = Section(11)
weapons = csv.DictReader(weapons_csv) print(weapons.fieldnames) # This should be the general format: # { # "name":"Padded Armor", # "equip_type":"light-armor", # "base_cost":5, # "weight":8, # "base_attributes":{ # "armor_class":11, # "stealth":"disadvantage" # } # } for row in weapons: new_equip = Equipment() payload = { "name": row["name"], "equip_type": row["equip_type"], "base_cost": float(row["base_cost"]), "weight": int(row["weight"]), "base_attributes": { "damage": row["damage"], "damage_type": row["damage_type"], "properties": row["properties"], "range": row["range"], "hands": row["hands"], }, } is_valid = new_equip.create(payload)
armor = csv.DictReader(armor_csv) print(armor.fieldnames) # This should be the general format: # { # "name":"Padded Armor", # "equip_type":"light-armor", # "base_cost":5, # "weight":8, # "base_attributes":{ # "armor_class":11, # "stealth":"disadvantage" # } # } for row in armor: new_armor = Equipment() payload = { "name": row["name"], "equip_type": row["equip_type"], "base_cost": float(row["base_cost"]), "weight": int(row["weight"]), "base_attributes": { "armor_class": row["armor_class"], "stealth": row["stealth"], }, } is_valid = new_armor.create(payload) if is_valid: response = database.insert(new_armor) print(response)