File: display_service.py

package info (click to toggle)
kylin-display-switch 1.0.3-1
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 216 kB
  • sloc: python: 540; sh: 4; makefile: 2
file content (122 lines) | stat: -rw-r--r-- 4,419 bytes parent folder | download
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
#!/usr/bin/python
# -*- coding: utf-8 -*-

import sys
import subprocess
from enums import DisplayMode


class DisplayService:

    current_display_mode = DisplayMode.MODE_ONLY_ORI

    def switch_display(self, target_mode):
        XRANDR_STATUS = "xrandr -q"

        handle = subprocess.Popen(XRANDR_STATUS, shell=True, stdout=subprocess.PIPE)
        handle.wait()
        res = handle.communicate()[0]
        res = res.decode()
        lines = res.splitlines()

        monitors = []
        modes = {}
        modes_one = []
        current_monitor = None
        flag = False

        for line in lines:
            if(line.find(" connected") != -1):
                monitor_name = line[: line.find(" ")]

                monitors.append(monitor_name)

                # find monitor again,add to dict
                if(flag == True):
                    modes[current_monitor] = modes_one

                current_monitor = monitor_name
                flag = True

                modes_one = []
            else:
                if(flag == True and line.startswith(" ") == True):
                    modes_one.append(line.strip().split(" ")[0])

        # add last monitor's mode
        modes[current_monitor] = modes_one

        print("found monitors :", monitors)

        if(len(monitors) < 2):
            handle = subprocess.Popen("xrandr --output " + monitors[0] + " --auto", shell=True, stdout=subprocess.PIPE)
            handle.wait()
            self.current_display_mode = DisplayMode.MODE_ONLY_ORI
            print("only one monitor, do nothing.")
            return (self.current_display_mode, False)

        else:
            modes_0 = modes[monitors[0]]
            modes_1 = modes[monitors[1]]
            MAX_MODE_MONITOR_0 = modes_0[0]
            MAX_MODE_MONITOR_1 = modes_1[0]
            BEST_CLONE_MODE = None

            is_find = False
            for mode_0 in modes_0:
                for mode_1 in modes_1:
                    if(mode_0 == mode_1):
                        BEST_CLONE_MODE = mode_0
                        is_find = True
                        break
                if(is_find == True):
                    break

            print("max display mode :", MAX_MODE_MONITOR_0, ",", MAX_MODE_MONITOR_1)
            print("best clone mode :", BEST_CLONE_MODE)

            XRANDR_ORIONLY = "xrandr --output " + monitors[0] + " --auto --output " + monitors[1] + " --off"
            XRANDR_CLONE = "xrandr --output " + monitors[0] + " --mode " + BEST_CLONE_MODE + " --output " + monitors[1] + " --mode " + BEST_CLONE_MODE + " --same-as " + monitors[0]
            XRANDR_EXTEND = "xrandr --output " + monitors[0] + " --auto --output " + monitors[1] + " --right-of " + monitors[0] + " --auto"
            XRANDR_OUTONLY = "xrandr --output " + monitors[0] + " --off --output " + monitors[1] + " --auto"

            try:
                if(target_mode == DisplayMode.MODE_ONLY_ORI):
                    handle = subprocess.Popen(XRANDR_ORIONLY, shell=True, stdout=subprocess.PIPE)
                    handle.wait()
                    self.current_display_mode = DisplayMode.MODE_ONLY_ORI

                elif(target_mode == DisplayMode.MODE_CLONE):
                    handle = subprocess.Popen(XRANDR_CLONE, shell=True, stdout=subprocess.PIPE)
                    handle.wait()
                    self.current_display_mode = DisplayMode.MODE_CLONE

                elif(target_mode == DisplayMode.MODE_EXTEND):
                    handle = subprocess.Popen(XRANDR_EXTEND, shell=True, stdout=subprocess.PIPE)
                    handle.wait()
                    handle = subprocess.Popen(XRANDR_EXTEND, shell=True, stdout=subprocess.PIPE)
                    handle.wait()
                    self.current_display_mode = DisplayMode.MODE_EXTEND

                elif(target_mode == DisplayMode.MODE_ONLY_OUT):
                    handle = subprocess.Popen(XRANDR_OUTONLY, shell=True, stdout=subprocess.PIPE)
                    handle.wait()
                    self.current_display_mode = DisplayMode.MODE_ONLY_OUT

                return (self.current_display_mode, True)
            except Exception as e:
                print(e)
                return (self.current_display_mode, False)


def main():
    # app = QApplication(sys.argv)

    w = DisplayService()
    w.switch_display(1)

    # sys.exit(app.exec_())


if __name__ == '__main__':
    main()