|
1 | 1 | # Handle misc CLI menu options |
2 | 2 | # Copyright (C) 2020-2021, Dhinak G, Mykola Grymalyuk |
3 | 3 | from __future__ import print_function |
| 4 | +import subprocess |
4 | 5 |
|
5 | 6 | from Resources import ModelArray, Constants, Utilities |
6 | 7 |
|
@@ -378,3 +379,33 @@ def disable_cpufriend(self): |
378 | 379 | self.constants.disallow_cpufriend = False |
379 | 380 | else: |
380 | 381 | print("Invalid option") |
| 382 | + |
| 383 | + def set_seedutil(self): |
| 384 | + Utilities.cls() |
| 385 | + Utilities.header(["Set SeedUtil Status"]) |
| 386 | + print("""Used for setting OS Update Preferences |
| 387 | +
|
| 388 | +Valid options: |
| 389 | +1. Public Release Seed (Default) |
| 390 | +2. Public Beta Seed |
| 391 | +3. Developer Beta Seed |
| 392 | +4. Check SeedUtil's current status |
| 393 | + """) |
| 394 | + |
| 395 | + change_menu = input("Set update status(Press [ENTER] to exit): ") |
| 396 | + if change_menu == "1": |
| 397 | + subprocess.run(["sudo", "/System/Library/PrivateFrameworks/Seeding.framework/Versions/A/Resources/seedutil", "unenroll"], stdout=subprocess.PIPE).stdout.decode().strip().encode() |
| 398 | + elif change_menu == "2": |
| 399 | + subprocess.run(["sudo", "/System/Library/PrivateFrameworks/Seeding.framework/Versions/A/Resources/seedutil", "unenroll"], stdout=subprocess.PIPE).stdout.decode().strip().encode() |
| 400 | + subprocess.run(["sudo", "/System/Library/PrivateFrameworks/Seeding.framework/Versions/A/Resources/seedutil", "enroll", "PublicSeed"], stdout=subprocess.PIPE).stdout.decode().strip().encode() |
| 401 | + elif change_menu == "3": |
| 402 | + subprocess.run(["sudo", "/System/Library/PrivateFrameworks/Seeding.framework/Versions/A/Resources/seedutil", "unenroll"], stdout=subprocess.PIPE).stdout.decode().strip().encode() |
| 403 | + subprocess.run(["sudo", "/System/Library/PrivateFrameworks/Seeding.framework/Versions/A/Resources/seedutil", "enroll", "DeveloperSeed"], stdout=subprocess.PIPE).stdout.decode().strip().encode() |
| 404 | + elif change_menu == "4": |
| 405 | + result = subprocess.run(["sudo", "/System/Library/PrivateFrameworks/Seeding.framework/Versions/A/Resources/seedutil", "current"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT) |
| 406 | + result = [i.partition(":")[2] for i in result.stdout.decode().split("\n") if "Currently enrolled in" in i][0] |
| 407 | + print(f"SeedUtil Current Status: {result}") |
| 408 | + input("\nPress [ENTER] to continue") |
| 409 | + self.set_seedutil() |
| 410 | + else: |
| 411 | + print("Returning to main menu") |
0 commit comments