40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
|
# Generated by Django 4.2.2 on 2024-03-11 15:19
|
||
|
|
import os
|
||
|
|
|
||
|
|
from django.db import migrations, models
|
||
|
|
|
||
|
|
|
||
|
|
class Migration(migrations.Migration):
|
||
|
|
dependencies = [
|
||
|
|
('hostadmin', '0002_importedidentifiersets'),
|
||
|
|
]
|
||
|
|
|
||
|
|
def calculate_hash(apps, schema_editor):
|
||
|
|
from hostadmin.models import ImportedIdentifierSets
|
||
|
|
for identifier_set in ImportedIdentifierSets.objects.all():
|
||
|
|
if not identifier_set.hash:
|
||
|
|
print("update", identifier_set.name)
|
||
|
|
filename = "shared_data/" + identifier_set.name.strip('git:') + ".json"
|
||
|
|
if not os.path.exists(filename):
|
||
|
|
continue
|
||
|
|
from hashlib import sha256
|
||
|
|
with open(filename, 'r') as file:
|
||
|
|
data = file.read()
|
||
|
|
identifier_set.hash = sha256(data.encode()).hexdigest()
|
||
|
|
identifier_set.save()
|
||
|
|
|
||
|
|
operations = [
|
||
|
|
migrations.AddField(
|
||
|
|
model_name='importedidentifiersets',
|
||
|
|
name='hash',
|
||
|
|
field=models.CharField(blank=True, max_length=255, null=True),
|
||
|
|
|
||
|
|
),
|
||
|
|
migrations.RunPython(calculate_hash),
|
||
|
|
migrations.AlterField(
|
||
|
|
model_name='importedidentifiersets',
|
||
|
|
name='hash',
|
||
|
|
field=models.CharField(max_length=255, unique=True),
|
||
|
|
),
|
||
|
|
]
|