You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
moonstone/vanilla_first_setup/main.py

118 lines
3.6 KiB
Python

# main.py
#
# Copyright 2022 mirkobrombin
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundationat version 3 of the License.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import os
import gi
import sys
import logging
from gettext import gettext as _
gi.require_version('Gtk', '4.0')
gi.require_version('Adw', '1')
gi.require_version('Vte', '3.91')
from gi.repository import Gtk, Gdk, Gio, GLib, Adw, Vte, Pango
from vanilla_first_setup.window import VanillaWindow
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger("FirstSetup::Main")
class FirstSetupApplication(Adw.Application):
"""The main application singleton class."""
def __init__(self):
super().__init__(application_id='org.vanillaos.FirstSetup',
flags=Gio.ApplicationFlags.HANDLES_COMMAND_LINE)
self.post_script = None
self.user = os.environ.get("USER")
self.new_user = False
self.create_action('quit', self.close, ['<primary>q'])
self.__register_arguments()
def __register_arguments(self):
"""Register the command line arguments."""
self.add_main_option(
"run-post-script",
ord("p"),
GLib.OptionFlags.NONE,
GLib.OptionArg.STRING,
_("Run a post script"),
None
)
self.add_main_option(
"new-user",
ord("p"),
GLib.OptionFlags.NONE,
GLib.OptionArg.NONE,
_("Run as a new user"),
None
)
def do_command_line(self, command_line):
"""Handle command line arguments."""
options = command_line.get_options_dict()
if options.contains("run-post-script"):
logger.info("Running post script")
self.post_script = options.lookup_value("run-post-script").get_string()
if options.contains("new-user"):
logger.info("Running as a new user")
self.user = None
self.new_user = True
self.activate()
def do_activate(self):
win = self.props.active_window
if not win:
win = VanillaWindow(application=self, post_script=self.post_script, user=self.user, new_user=self.new_user)
win.present()
def create_action(self, name, callback, shortcuts=None):
"""Add an application action.
Args:
name: the name of the action
callback: the function to be called when the action is
activated
shortcuts: an optional list of accelerators
"""
action = Gio.SimpleAction.new(name, None)
action.connect("activate", callback)
self.add_action(action)
if shortcuts:
self.set_accels_for_action(f"app.{name}", shortcuts)
def close(self, *args):
"""Close the application."""
self.quit()
def main(version):
"""The application's entry point."""
if os.environ.get("USERNAME") in ["ubuntu", "vanillaos", "vanilla-os"]:
logging.warning("Running in Live mode, closing...")
sys.exit(0)
app = FirstSetupApplication()
return app.run(sys.argv)