NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-02 05:46:25 -06:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Refactor Lookup subclasses to be more DRY

Browse Source
This commit is contained in:
Brian Tiemann 2024-09-09 14:15:05 -04:00
parent ad3df60fdd
commit fcfbb1e67f
1 changed files with 29 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
netbox/ipam/lookups.py
View File

@ -154,44 +154,49 @@ class NetHostContained(Lookup):
return 'CAST(HOST(%s) AS INET) <<= %s' % (lhs, rhs), params
class NetHostGreaterThan(Lookup):
class NetHostComparison(Lookup):
@property
def comparison_sql(self):
raise NotImplementedError
def as_sql(self, qn, connection):
lhs, lhs_params = self.process_lhs(qn, connection)
rhs, rhs_params = self.process_rhs(qn, connection)
params = lhs_params + rhs_params
return self.comparison_sql % (lhs, rhs), params
class NetHostGreaterThan(NetHostComparison):
lookup_name = 'net_host_gt'
def as_sql(self, qn, connection):
lhs, lhs_params = self.process_lhs(qn, connection)
rhs, rhs_params = self.process_rhs(qn, connection)
params = lhs_params + rhs_params
return 'CAST(HOST(%s) AS INET) > INET %s' % (lhs, rhs), params
@property
def comparison_sql(self):
return 'CAST(HOST(%s) AS INET) > INET %s'
class NetHostLessThan(Lookup):
class NetHostLessThan(NetHostComparison):
lookup_name = 'net_host_lt'
def as_sql(self, qn, connection):
lhs, lhs_params = self.process_lhs(qn, connection)
rhs, rhs_params = self.process_rhs(qn, connection)
params = lhs_params + rhs_params
return 'CAST(HOST(%s) AS INET) < INET %s' % (lhs, rhs), params
@property
def comparison_sql(self):
return 'CAST(HOST(%s) AS INET) < INET %s'
class NetHostGreaterThanOrEqual(Lookup):
class NetHostGreaterThanOrEqual(NetHostComparison):
lookup_name = 'net_host_gte'
def as_sql(self, qn, connection):
lhs, lhs_params = self.process_lhs(qn, connection)
rhs, rhs_params = self.process_rhs(qn, connection)
params = lhs_params + rhs_params
return 'CAST(HOST(%s) AS INET) >= INET %s' % (lhs, rhs), params
@property
def comparison_sql(self):
return 'CAST(HOST(%s) AS INET) >= INET %s'
class NetHostLessThanOrEqual(Lookup):
class NetHostLessThanOrEqual(NetHostComparison):
lookup_name = 'net_host_lte'
def as_sql(self, qn, connection):
lhs, lhs_params = self.process_lhs(qn, connection)
rhs, rhs_params = self.process_rhs(qn, connection)
params = lhs_params + rhs_params
return 'CAST(HOST(%s) AS INET) <= INET %s' % (lhs, rhs), params
@property
def comparison_sql(self):
return 'CAST(HOST(%s) AS INET) <= INET %s'
class NetFamily(Transform):