mirror of
https://github.com/aaif-goose/goose.git
synced 2026-07-03 14:10:03 +02:00
Tyler Longwell
37 lines
858 B
Python
37 lines
858 B
Python
"""Optional extension system for advanced users"""
|
|
|
|
from abc import ABC, abstractmethod
|
|
from typing import TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from ..client import GooseClient
|
|
|
|
|
|
class Extension(ABC):
|
|
"""Base class for extensions"""
|
|
|
|
def __init__(self, client: "GooseClient"):
|
|
"""Initialize the extension with a client instance.
|
|
|
|
Args:
|
|
client: The GooseClient instance to extend
|
|
"""
|
|
self.client = client
|
|
|
|
@abstractmethod
|
|
def install(self):
|
|
"""Install the extension.
|
|
|
|
This method should modify the client instance to add
|
|
the extension's functionality.
|
|
"""
|
|
pass
|
|
|
|
def uninstall(self):
|
|
"""Uninstall the extension.
|
|
|
|
Optional method to remove the extension's modifications.
|
|
Default implementation does nothing.
|
|
"""
|
|
pass
|