from dataclasses import dataclass
from dataclasses_json import dataclass_json

@dataclass_json
@dataclass
class BaseConnection:
    code: str

    # Called by: ConnectionController
    # it uses it for basic type checks on wireguard code
    def needs_wireguard_configuration(self):
        return self.code == 'wireguard'

    # Called By SubscriptionPlan
    def is_session_connection(self):
        return type(self).__name__ == 'SessionConnection'

    # Not called. Dead code
    def is_system_connection(self):
        return type(self).__name__ == 'SystemConnection'