ubuntu/createlts.py at master · bdaloukas/ubuntu · GitHub

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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#!/usr/bin/python
#-*- coding: utf-8 -*-
# Copyright (C) 2016 Vasilis Daloukas <bdaloukas@gmail.com>
# License GNU GPL version 3 or newer <http://gnu.org/licenses/gpl.html>

import os, shutil

def scan_network():
    s = os.system("arp -a > a.txt")
    f = open('a.txt', "r")
    for line in f:
        pos = line.find( ".local")
        if pos == -1:
            continue
        computer = line[ 0:pos]
        if computer[ 0:4] != "ltsp":
            continue
        pos = line.find( ')')
        if pos == -1:
            continue
        line2 = line[ pos+1:]
        pos = line2.find( ' ')
        line2 = line2[ pos+1:]
        pos = line2.find( ' ')
        if pos == -1:
            continue
        line2 = line2[ pos+1:]
        pos = line2.find( ' ')
        if pos == -1:
            continue
        mac = line2[ 0:pos].upper()
        UpdateLTS( mac, computer)

def ComputeUserName( computer):
    s = os.system("who > b.txt")
    f = open('b.txt', "r")
    search = "(" + computer + ".local)"
    for line in f:
        if line.find( search) == -1:
            continue
        pos = line.find( " ")
        if pos == -1:
            continue
        username = line[ 0:pos]
        return username
    print "problem at " + computer
    return ""

def UpdateLTS(mac, computer):
    print "computer=" + computer + "  mac=" + mac
    filename = "/var/lib/tftpboot/ltsp/i386/lts.conf"
    temp_file = filename + ".tmp"
    search = "[" + mac + "]"
    lensearch = len( search)
    with open(filename, "r") as input_file:
        for line in input_file:
            #print "search=" + search + " l=" + line[ 0:lensearch]
            if line[ 0:lensearch] == search:
                return
            #text_file.write( "#!/bin/sh\n")
    print "must change"
    username = ComputeUserName( computer)
    if username == "":
        return
    with open(filename, "r") as input_file:
        with open(temp_file, "w") as text_file:
            for line in input_file:
                text_file.write( line)
            text_file.write( search + "\n")
            print "username=" + username
            text_file.write( "HOSTNAME=" + username + "\n")
            text_file.write( "LDM_USERNAME=" + username + "\n")
            text_file.write( "LDM_PASSWORD=" + username + "\n")
            text_file.write( "\n")

    bak_file = filename + ".bak"
    shutil.copyfile( filename, bak_file)
    shutil.copyfile( temp_file, filename)

def CheckLTS():
    filename = "/var/lib/tftpboot/ltsp/i386/lts.conf"
    found_XRANDR_MODE_0 = 0

    found_XRANDR_RATE_0 = 0
    with open(filename, "r") as input_file:
        for line in input_file:
            if line[ 0:len("XRANDR_MODE_0")] == "XRANDR_MODE_0" :
                found_XRANDR_MODE_0 = 1
            if line[ 0:len("XRANDR_RATE_0")] == "XRANDR_RATE_0" :
                found_XRANDR_RATE_0 = 1
            if line[ 0:len("FAT_RAM_THRESHOLD")] == "FAT_RAM_THRESHOLD":
                pos = line.find( "=")
                memory = int( line[ pos+1:])
                if memory < 600:
                    print "Καλύτερα είναι το FAT_RAM_THRESHOLD να γίνει 700"
    if found_XRANDR_MODE_0 == 0:
        print "Λείπει το XRANDR_MODE_0=1024x768 στο [OLD_MONITOR]"
    if found_XRANDR_RATE_0 == 0:
        print "Λείπει το XRANDR_RATE_0=85 στο [OLD_MONITOR]"

scan_network()
CheckLTS()
#print ret