0UT1S's

# testing/pickleable.py
# Copyright (C) 2005-2025 the SQLAlchemy authors and contributors
# <see AUTHORS file>
#
# This module is part of SQLAlchemy and is released under
# the MIT License: https://www.opensource.org/licenses/mit-license.php
# mypy: ignore-errors

"""Classes used in pickling tests, need to be at the module level for
unpickling.
"""

from __future__ import annotations

from .entities import ComparableEntity
from ..schema import Column
from ..types import String

class User(ComparableEntity):
    pass

class Order(ComparableEntity):
    pass

class Dingaling(ComparableEntity):
    pass

class EmailUser(User):
    pass

class Address(ComparableEntity):
    pass

# TODO: these are kind of arbitrary....
class Child1(ComparableEntity):
    pass

class Child2(ComparableEntity):
    pass

class Parent(ComparableEntity):
    pass

class Screen:
    def __init__(self, obj, parent=None):
        self.obj = obj
        self.parent = parent

class Mixin:
    email_address = Column(String)

class AddressWMixin(Mixin, ComparableEntity):
    pass

class Foo:
    def __init__(self, moredata, stuff="im stuff"):
        self.data = "im data"
        self.stuff = stuff
        self.moredata = moredata

__hash__ = object.__hash__

def __eq__(self, other):
        return (
            other.data == self.data
            and other.stuff == self.stuff
            and other.moredata == self.moredata
        )

class Bar:
    def __init__(self, x, y):
        self.x = x
        self.y = y

__hash__ = object.__hash__

def __eq__(self, other):
        return (
            other.__class__ is self.__class__
            and other.x == self.x
            and other.y == self.y
        )

def __str__(self):
        return "Bar(%d, %d)" % (self.x, self.y)

class OldSchool:
    def __init__(self, x, y):
        self.x = x
        self.y = y

def __eq__(self, other):
        return (
            other.__class__ is self.__class__
            and other.x == self.x
            and other.y == self.y
        )

class OldSchoolWithoutCompare:
    def __init__(self, x, y):
        self.x = x
        self.y = y

class BarWithoutCompare:
    def __init__(self, x, y):
        self.x = x
        self.y = y

def __str__(self):
        return "Bar(%d, %d)" % (self.x, self.y)

class NotComparable:
    def __init__(self, data):
        self.data = data

def __hash__(self):
        return id(self)

def __eq__(self, other):
        return NotImplemented

def __ne__(self, other):
        return NotImplemented

class BrokenComparable:
    def __init__(self, data):
        self.data = data

def __hash__(self):
        return id(self)

def __eq__(self, other):
        raise NotImplementedError

def __ne__(self, other):
        raise NotImplementedError