mirror of https://github.com/zulip/zulip.git
33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
|
from typing import List, Tuple, Union
|
||
|
|
|
||
|
|
from django.db import models
|
||
|
|
from django.db.backends.base.base import BaseDatabaseWrapper
|
||
|
|
from django.db.models.sql.compiler import SQLCompiler
|
||
|
|
from typing_extensions import override
|
||
|
|
|
||
|
|
|
||
|
|
@models.Field.register_lookup
|
||
|
|
class AndZero(models.Lookup[int]):
|
||
|
|
lookup_name = "andz"
|
||
|
|
|
||
|
|
@override
|
||
|
|
def as_sql(
|
||
|
|
self, compiler: SQLCompiler, connection: BaseDatabaseWrapper
|
||
|
|
) -> Tuple[str, List[Union[str, int]]]: # nocoverage # currently only used in migrations
|
||
|
|
lhs, lhs_params = self.process_lhs(compiler, connection)
|
||
|
|
rhs, rhs_params = self.process_rhs(compiler, connection)
|
||
|
|
return f"{lhs} & {rhs} = 0", lhs_params + rhs_params
|
||
|
|
|
||
|
|
|
||
|
|
@models.Field.register_lookup
|
||
|
|
class AndNonZero(models.Lookup[int]):
|
||
|
|
lookup_name = "andnz"
|
||
|
|
|
||
|
|
@override
|
||
|
|
def as_sql(
|
||
|
|
self, compiler: SQLCompiler, connection: BaseDatabaseWrapper
|
||
|
|
) -> Tuple[str, List[Union[str, int]]]: # nocoverage # currently only used in migrations
|
||
|
|
lhs, lhs_params = self.process_lhs(compiler, connection)
|
||
|
|
rhs, rhs_params = self.process_rhs(compiler, connection)
|
||
|
|
return f"{lhs} & {rhs} != 0", lhs_params + rhs_params
|