ods

from contextlib import nullcontext
from typing import ContextManager, Optional, Union

import pytest


def maybe_raises(
    exception: Union[type[BaseException], tuple[type[BaseException], ...]]
) -> ContextManager[Optional[pytest.ExceptionInfo[BaseException]]]:
    if exception is None:

ods

import asyncio
from contextlib import AsyncExitStack, ExitStack
import contextvars
import functools
import inspect
import sys


__all__ = ['exit_stack']

ods

import time, asyncio, hashlib


def sync_cpu_bound():
    s = bytes(range(256))
    for i in range(100000):
        hashlib.sha256(s)

start = time.time()
sync_cpu_bound()

ods

"""
Benchmark for PR to serialize execution of SQL in asyncpg.
See: https://github.com/MagicStack/asyncpg/pull/367
"""

import asyncio
from asyncio import Lock


class _Atomic:

ods

from sqlalchemy import Column, Integer, String, ForeignKey, create_engine
from sqlalchemy.orm import relationship, sessionmaker
from sqlalchemy.ext.declarative import declarative_base


Base = declarative_base()


class DocLink(Base):
    __tablename__ = 'DocLink'

ods

from sqlalchemy import Column, Integer, ForeignKey
from sqlalchemy.orm import relationship
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class Link(Base):
    __tablename__ = 'Link'
    id = Column(Integer, primary_key=True)
    doc_id = Column(Integer, ForeignKey('Doc.id'), nullable=False)

ods

import sphinx_xmlpipe2

pipe = sphinx_xmlpipe2.Pipe(
    fields = ['title', 'description', 'body'],
    attrs = [sphinx_xmlpipe2.Timestamp('date'), sphinx_xmlpipe2.Int8('type')],
)

with pipe:
    for doc in docs:
        pipe.add(

ods

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String, ForeignKey, create_engine
from sqlalchemy.orm import sessionmaker, relationship
from sqlalchemy.ext.orderinglist import ordering_list

Base = declarative_base()


class Parent(Base):
    __tablename__ = 'parents'

ods

registry = Registry()

ReplicatedModel = registry.multidb(
    admin=AdminBase,
    front=FrontBase,
)


@registry.model(ReplicatedModel)
def Media(models):

ods

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, ForeignKey, create_engine
from sqlalchemy.orm import sessionmaker, Session, relationship

Base = declarative_base()


class AB(Base):
    __tablename__ = 'AB'
    a_id = Column(ForeignKey('A.id'), primary_key=True)