Infra/lemur
5
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #2797 from castrapel/get_or_increase_name_simplify

Browse Source 
Make get_or_increase_name queries less demanding
This commit is contained in:
Curtis 2019-05-29 12:50:06 -07:00 committed by GitHub
commit 8bc23f6deb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lemur/certificates/models.py
View File

@ -71,21 +71,21 @@ def get_sequence(name):
def get_or_increase_name(name, serial):
certificates = Certificate.query.filter(
Certificate.name.ilike("{0}%".format(name))
).all()
certificates = Certificate.query.filter(Certificate.name == name).all()
if not certificates:
return name
serial_name = "{0}-{1}".format(name, hex(int(serial))[2:].upper())
certificates = Certificate.query.filter(
Certificate.name.ilike("{0}%".format(serial_name))
).all()
certificates = Certificate.query.filter(Certificate.name == serial_name).all()
if not certificates:
return serial_name
certificates = Certificate.query.filter(
Certificate.name.ilike("{0}%".format(serial_name))
).all()
ends = [0]
root, end = get_sequence(serial_name)
for cert in certificates:

3
lemur/tests/test_certificates.py
View File

@ -53,7 +53,8 @@ def test_get_or_increase_name(session, certificate):
== "test-cert-11111111-1-" + serial
)
cert2 = CertificateFactory(name="certificate1-" + serial)
CertificateFactory(name="certificate1")
CertificateFactory(name="certificate1-" + serial)
session.commit()
assert get_or_increase_name(