119 lines
5.1 KiB
Python
119 lines
5.1 KiB
Python
import asyncio
|
|
import time
|
|
from bleak import BleakScanner, BleakClient
|
|
|
|
lampAmount = 25
|
|
LampArray = {
|
|
"Center": [0],
|
|
"Up": [1,2,8,9,10,11,12,13,23,24],
|
|
"Down": [4,5,6,15,16,17,18,19,20,21],
|
|
"Left": [6,3,8,9,19,20,21,22,23,12],
|
|
"Right": [2,7,4,11,24,13,14,15,16,17],
|
|
"Inner": [1,2,3,4,5,6,7,8],
|
|
"Outer": [9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24]
|
|
}
|
|
colorArray = ["500000","005000","000050"]
|
|
colorArrayTest = ["050000","000500"]
|
|
|
|
def sectionInitializer():
|
|
result = [b'\x00\x00\x00'] * lampAmount
|
|
return result
|
|
|
|
def sectionSerializer(section, color):
|
|
result = sectionInitializer()
|
|
sectionResult = LampArray[section]
|
|
for pos in sectionResult:
|
|
result[pos] = int(color,16).to_bytes(3,'big')
|
|
return result
|
|
|
|
def posColorByte(pos, section, color):
|
|
intColor = int(color, 16)
|
|
result = int(intColor * section[pos])
|
|
print(result)
|
|
return result.to_bytes(3)
|
|
|
|
async def setOnUserInput(client, chars):
|
|
while True:
|
|
configZone = input("Input Zone [Up, Down, Left, Right, Inner, Outer]: ")
|
|
if configZone not in list(LampArray.keys()):
|
|
continue
|
|
configColor = input("Input Color [000000]: ")
|
|
configColorCenter = input("Input Color for Center Lamp [000000]: ")
|
|
await setLampToColor(client, chars, configZone, configColor, configColorCenter)
|
|
|
|
async def setLampToColor(bleClient, bleChars, section, color, colorCenter):
|
|
sectionSerial = sectionSerializer(section, color)
|
|
sectionSerial[0] = int(colorCenter,16).to_bytes(3,'big')
|
|
for char in bleChars:
|
|
lampPos = bleChars.index(char)
|
|
value = sectionSerial[lampPos]
|
|
print(f"Setting Lamp number {lampPos} to {value}")
|
|
await bleClient.write_gatt_char(char.uuid, value)
|
|
|
|
async def connect_to_ble_device(device_name):
|
|
print(f"Scanning for device: {device_name}...")
|
|
devices = await BleakScanner.discover()
|
|
global services
|
|
global client
|
|
|
|
target_device = None
|
|
for device in devices:
|
|
if device.name == device_name:
|
|
target_device = device
|
|
break
|
|
|
|
if target_device:
|
|
print(f"Found device: {target_device.name} ({target_device.address})")
|
|
async with BleakClient(target_device.address) as client:
|
|
if client.is_connected:
|
|
print(f"Connected to {target_device.name}")
|
|
# Now you can interact with services and characteristics
|
|
# Example: Read a characteristic (replace with actual UUID)
|
|
# battery_level = await client.read_gatt_char("00002a19-0000-1000-8000-00805f9b34fb")
|
|
# print(f"Battery Level: {int.from_bytes(battery_level, 'little')}%")
|
|
else:
|
|
print(f"Failed to connect to {target_device.name}")
|
|
exit(1)
|
|
services = sorted(client.services, key=lambda serv: serv.handle)
|
|
print("\n--- Services and Characteristics ---")
|
|
for service in services:
|
|
print(f"\nService: {service.uuid} (Handle: {service.handle}) - {service.description}")
|
|
# Get all characteristics for the current service
|
|
characteristics = service.characteristics
|
|
|
|
# Sort the characteristics by handle
|
|
sorted_characteristics = sorted(characteristics, key=lambda char: char.handle)
|
|
print("Characteristics sorted by handle:")
|
|
for char in sorted_characteristics:
|
|
print(f" Characteristic: {char.uuid} (Handle: {char.handle})")
|
|
print(f" Description: {char.description}")
|
|
print(f" Properties: {char.properties}")
|
|
|
|
if "read" in char.properties:
|
|
try:
|
|
await client.write_gatt_char(char.uuid, bytes.fromhex("000005"))
|
|
time.sleep(1)
|
|
value = await client.read_gatt_char(char.uuid)
|
|
print(f" Value: {value.hex()} (raw bytes)")
|
|
except Exception as e:
|
|
print(f" Could not read characteristic {char.uuid}: {e}")
|
|
else:
|
|
print(" (Read not supported)")
|
|
|
|
# for color in colorArray:
|
|
# await setLampToColor(client, service.characteristics, Up, color)
|
|
# await setLampToColor(client, service.characteristics, Down, color)
|
|
# await setLampToColor(client, service.characteristics, Left, color)
|
|
# await setLampToColor(client, service.characteristics, Right, color)
|
|
# await setLampToColor(client, service.characteristics, Inner, color)
|
|
# await setLampToColor(client, service.characteristics, Outer, color)
|
|
if service.handle != 1:
|
|
await setOnUserInput(client, sorted_characteristics)
|
|
|
|
else:
|
|
print(f"Device '{device_name}' not found.")
|
|
|
|
if __name__ == "__main__":
|
|
# Replace 'YourDeviceName' with the actual name of your BLE device
|
|
asyncio.run(connect_to_ble_device("Pupilometer LED Billboard"))
|