1
|
from sqlalchemy.orm import Session
|
2
|
|
3
|
from . import models, schemas
|
4
|
|
5
|
|
6
|
def get_device(db: Session, device_id: int):
|
7
|
return db.query(models.Device).filter(models.Device.id == device_id).first()
|
8
|
|
9
|
def get_devices(db: Session, skip: int = 0, limit: int = 100):
|
10
|
return db.query(models.Device).offset(skip).limit(limit).all()
|
11
|
|
12
|
|
13
|
def create_device(db: Session, device: schemas.DeviceCreate):
|
14
|
db_device = models.Device(vendor_id=device.vendor_id, product_id=device.product_id)
|
15
|
db.add(db_device)
|
16
|
db.commit()
|
17
|
db.refresh(db_device)
|
18
|
return db_device
|
19
|
|
20
|
|
21
|
def get_logs(db: Session, skip: int = 0, limit: int = 100):
|
22
|
return db.query(models.USBLog).offset(skip).limit(limit).all()
|
23
|
|
24
|
def get_log(db: Session, device_id: int, skip: int=0, limit: int=100):
|
25
|
return db.query(models.USBLog).filter(models.USBLog.device_id == device_id).offset(skip).limit(limit).all()
|
26
|
|
27
|
def create_device_logs(db: Session, item: schemas.USBLogCreate, dev_id: int):
|
28
|
db_log = models.USBLog(**item.dict(), device_id=dev_id)
|
29
|
db.add(db_log)
|
30
|
db.commit()
|
31
|
db.refresh(db_log)
|
32
|
return db_log
|
33
|
|