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
|
#!/usr/bin/python3
import sys
import sqlite3
usageInfo = f"usage: {sys.argv[0]}\n"
usageInfo += "Reads alt-name data from a file, and adds it to the 'names' table.\n"
usageInfo += "The file is expected to have lines of the form: nodeName|altName|prefAlt\n"
usageInfo += " These correspond to entries in the 'names' table. 'prefAlt' should\n"
usageInfo += " be 1 or 0. A line may specify name1|name1|1, which causes the node\n"
usageInfo += " to have no preferred alt-name.\n"
if len(sys.argv) > 1:
print(usageInfo, file=sys.stderr)
sys.exit(1)
dbFile = "data.db"
pickedNamesFile = "pickedNames.txt"
# Open db
dbCon = sqlite3.connect(dbFile)
dbCur = dbCon.cursor()
# Iterate through picked-names file
with open(pickedNamesFile) as file:
for line in file:
# Get record data
(nodeName, altName, prefAlt) = line.lower().rstrip().split("|")
prefAlt = int(prefAlt)
# Remove any existing preferred-alt status
if prefAlt == 1:
query = "SELECT name, alt_name FROM names WHERE name = ? AND pref_alt = 1"
row = dbCur.execute(query, (nodeName,)).fetchone()
if row != None and row[1] != altName:
print(f"Removing pref-alt status from alt-name {row[1]} for {nodeName}")
dbCur.execute("UPDATE names SET pref_alt = 0 WHERE name = ? AND alt_name = ?", row)
# Check for an existing record
if nodeName == altName:
continue
query = "SELECT name, alt_name, pref_alt FROM names WHERE name = ? AND alt_name = ?"
row = dbCur.execute(query, (nodeName, altName)).fetchone()
if row == None:
print(f"Adding record for alt-name {altName} for {nodeName}")
dbCur.execute("INSERT INTO names VALUES (?, ?, ?, 'picked')", (nodeName, altName, prefAlt))
else:
# Update existing record
if row[2] != prefAlt:
print(f"Updating record for alt-name {altName} for {nodeName}")
dbCur.execute("UPDATE names SET pref_alt = ?, src = 'picked' WHERE name = ? AND alt_name = ?",
(prefAlt, nodeName, altName))
# Close db
dbCon.commit()
dbCon.close()
|
