blob: db194e28b544ce97eaf591c1cd5b8ca89ef60af2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
#!/usr/bin/python
import sbhs
import os
import sys
def scan_machines():
# erase the old map_machine_ids.txt file
try:
file('map_machine_ids.txt', 'w').close()
except:
print 'Failed to create machine map file file'
sys.exit(1)
# open the map_machine_ids file for writing
try:
map_machine_file = file('map_machine_ids.txt', 'w')
except:
print 'Failed to create machine map file file'
sys.exit(1)
# get list of device file names that start with ttyUSB* in the /dev folder
device_files = []
device_files += [each for each in os.listdir('/dev') if each.startswith('ttyUSB')]
# if no device filename found then exit
if not device_files:
print 'No USB device found in /dev folder'
sys.exit(1)
for device in device_files:
s = sbhs.Sbhs()
# getting the number from the device filename
dev_id = device[6:]
try:
dev_id = int(dev_id)
except:
print 'Invalid device name /dev/%s' % device
continue
# connect to device
res = s.connect_device(dev_id)
if not res:
print 'Cannot connect to /dev/%s' % device
s.disconnect()
continue
# get the machine id from the device
machine_id = s.getMachineId()
if machine_id < 0:
print 'Cannot get machine id from /dev/%s' % device
s.disconnect()
continue
print 'Found SBHS device /dev/%s with machine id %d' % (device, machine_id)
map_str = "%d=/dev/%s\n" % (machine_id, device)
map_machine_file.write(map_str)
#if __name__ == 'main':
#print 'Done. Exiting...'
map_machine_file.close()
#sys.exit(1)
|
