Fix MQTT client import error

Author: Bagus-DevLab

app/api/v1/devices.py

@@ -3,15 +3,17 @@
 from pydantic import BaseModel
 from typing import Optional
 
-# --- UPDATE IMPORT (Sesuai Struktur Baru) ---
-from app.core.database import get_db          # <-- Folder db
-from app.models.device import SensorLog     # <-- Sesuaikan nama class device kamu (Device/SensorLog?)
-# Cek model kamu: Kalau nama classnya Device, ganti SensorLog jadi Device di baris atas
-from app.mqtt.client import publish_message # <-- Kita pakai helper publish kalau ada, atau client langsung
+# --- IMPORT DATABASE ---
+from app.core.database import get_db
+from app.models.device import Device 
 
-# --- IMPORT RATE LIMITER REDIS ---
+# --- IMPORT RATE LIMITER ---
 from fastapi_limiter.depends import RateLimiter
 
+# --- PERBAIKAN IMPORT MQTT ---
+# Kita import object 'client' langsung, lalu kita alias-kan jadi 'mqtt_client'
+from app.mqtt.client import client as mqtt_client 
+
 router = APIRouter()
 
 # --- SCHEMA ---
@@ -24,25 +26,18 @@ class AutoRegisterSchema(BaseModel):
     pin_code: str
     factory_secret: str
 
-# Kita butuh model Device (Definisikan ulang kalau belum import)
-# Asumsi kamu punya model Device di app/models/device.py
-from app.models.device import Device # Pastikan ini ada
-
 # --- 1. ENDPOINT CLAIM (User) ---
-# Rate Limit: 5x per 60 detik
 @router.post("/claim", dependencies=[Depends(RateLimiter(times=5, seconds=60))])
 def claim_device(
     claim_data: UserClaimSchema, 
     db: Session = Depends(get_db)
-    # user_uid: str = Depends(get_current_user) # <-- Aktifkan ini nanti kalau AUTH firebase sudah dipasang
 ):
-    # Sementara kita hardcode user_uid dulu biar bisa tes tanpa login frontend
+    # Hardcode user sementara
     user_uid = "TEST_USER_UID_001" 
 
     clean_id = claim_data.device_id.strip().upper()
     clean_pin = claim_data.pin_code.strip()
 
-    # Cari Alat
     device = db.query(Device).filter(Device.device_id == clean_id).first()
 
     if not device:
@@ -53,11 +48,9 @@ def claim_device(
              return {"message": "Alat ini memang sudah punya kamu kok."}
         raise HTTPException(status_code=400, detail="Alat ini sudah dimiliki orang lain!")
 
-    # Cek PIN
     if device.pin_code != clean_pin:
         raise HTTPException(status_code=400, detail="PIN Salah!")
 
-    # Simpan Pemilik
     device.owner_uid = user_uid
     device.device_name = "Alat Baru Saya"
     db.commit()
@@ -66,16 +59,12 @@ def claim_device(
 
 
 # --- 2. ENDPOINT CONTROL RELAY ---
-# Rate Limit: 10x per 60 detik
 @router.post("/control-relay", dependencies=[Depends(RateLimiter(times=10, seconds=60))])
 def control_relay(
     device_id: str,
     state: str,
     db: Session = Depends(get_db)
 ):
-    # Hardcode user dulu
-    user_uid = "TEST_USER_UID_001"
-    
     clean_id = device_id.strip().upper()
 
     if state not in ["ON", "OFF"]:
@@ -86,16 +75,11 @@ def control_relay(
     if not device:
         raise HTTPException(status_code=404, detail="Alat tidak ditemukan.")
 
-    # Proteksi Kepemilikan (Nyalakan nanti kalau Auth siap)
-    # if device.owner_uid != user_uid:
-    #    raise HTTPException(status_code=403, detail="Bukan alat kamu!")
-
     topic = f"alat/{clean_id}/command"
     payload = f'{{"relay": "{state}"}}'
 
-    # Publish ke MQTT
-    # Pastikan kamu import client mqtt yg benar, atau pakai cara ini:
-    from app.mqtt.client import client as mqtt_client
+    # --- PERBAIKAN CARA PUBLISH ---
+    # Gunakan method .publish() milik object client
     mqtt_client.publish(topic, payload)
 
     return {"message": "Perintah dikirim", "topic": topic, "state": state}
@@ -123,7 +107,7 @@ def auto_register_device(
         device_id=clean_id,
         pin_code=clean_pin,
         device_name="New Device",
-        # is_active=True # Sesuaikan dengan model kamu
+        # is_active=True
     )
     db.add(new_device)
     db.commit()