from dataclasses import dataclass
from dataclasses_json import dataclass_json


@dataclass_json
@dataclass
class BaseConnection:
    code: str

    def needs_wireguard_configuration(self):
        return self.code == 'wireguard'