Merge item snapshot and item to be one model.

master
Tom Hacohen 5 years ago
parent 818bb8d70f
commit 0a3bb6f4bb

@ -0,0 +1,53 @@
# Generated by Django 3.0.3 on 2020-02-20 09:43
import django.core.validators
from django.db import migrations, models
import django.db.models.deletion
class Migration(migrations.Migration):
dependencies = [
('django_etesync', '0001_initial'),
]
operations = [
migrations.RemoveField(
model_name='collectionitemchunk',
name='itemSnapshot',
),
migrations.AddField(
model_name='collectionitem',
name='hmac',
field=models.CharField(default='', max_length=50),
preserve_default=False,
),
migrations.AddField(
model_name='collectionitemchunk',
name='items',
field=models.ManyToManyField(related_name='chunks', to='django_etesync.CollectionItem'),
),
migrations.AlterField(
model_name='collection',
name='uid',
field=models.CharField(db_index=True, max_length=44, validators=[django.core.validators.RegexValidator(message='Not a valid UID. Expected a 256bit base64url.', regex='[a-zA-Z0-9\\-_=]{44}')]),
),
migrations.AlterField(
model_name='collectionitem',
name='collection',
field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='items', to='django_etesync.Collection'),
),
migrations.AlterField(
model_name='collectionitem',
name='uid',
field=models.CharField(db_index=True, max_length=44, validators=[django.core.validators.RegexValidator(message='Not a valid UID. Expected a 256bit base64url.', regex='[a-zA-Z0-9\\-_=]{44}')]),
),
migrations.AlterField(
model_name='collectionitemchunk',
name='uid',
field=models.CharField(db_index=True, max_length=44, validators=[django.core.validators.RegexValidator(message='Not a valid UID. Expected a 256bit base64url.', regex='[a-zA-Z0-9\\-_=]{44}')]),
),
migrations.DeleteModel(
name='CollectionItemSnapshot',
),
]

@ -0,0 +1,18 @@
# Generated by Django 3.0.3 on 2020-02-20 09:47
from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
('django_etesync', '0002_auto_20200220_0943'),
]
operations = [
migrations.AddField(
model_name='collectionitem',
name='current',
field=models.BooleanField(default=True),
),
]

@ -0,0 +1,18 @@
# Generated by Django 3.0.3 on 2020-02-20 10:29
from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
('django_etesync', '0003_collectionitem_current'),
]
operations = [
migrations.AlterField(
model_name='collectionitem',
name='current',
field=models.BooleanField(db_index=True, default=True),
),
]

@ -12,6 +12,8 @@
# You should have received a copy of the GNU General Public License # You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>. # along with this program. If not, see <http://www.gnu.org/licenses/>.
from pathlib import Path
from django.db import models from django.db import models
from django.conf import settings from django.conf import settings
from django.core.validators import RegexValidator from django.core.validators import RegexValidator
@ -33,45 +35,41 @@ class Collection(models.Model):
def __str__(self): def __str__(self):
return self.uid return self.uid
@cached_property
def current_items(self):
return self.items.filter(current=True)
class CollectionItem(models.Model): class CollectionItem(models.Model):
uid = models.CharField(db_index=True, blank=False, null=False, uid = models.CharField(db_index=True, blank=False, null=False,
max_length=44, validators=[UidValidator]) max_length=44, validators=[UidValidator])
version = models.PositiveSmallIntegerField() version = models.PositiveSmallIntegerField()
encryptionKey = models.BinaryField(editable=True, blank=False, null=False) encryptionKey = models.BinaryField(editable=True, blank=False, null=False)
collection = models.ForeignKey(Collection, on_delete=models.CASCADE) collection = models.ForeignKey(Collection, related_name='items', on_delete=models.CASCADE)
hmac = models.CharField(max_length=50, blank=False, null=False)
current = models.BooleanField(db_index=True, default=True)
class Meta: class Meta:
unique_together = ('uid', 'collection') unique_together = ('uid', 'collection')
@cached_property
def content(self):
return self.snapshots.get(current=True)
def __str__(self): def __str__(self):
return self.uid return self.uid
class CollectionItemSnapshot(models.Model): def chunk_directory_path(instance, filename):
item = models.ForeignKey(CollectionItem, related_name='snapshots', on_delete=models.CASCADE) col = instance.itemSnapshot.item.collection
current = models.BooleanField(default=True) user_id = col.owner.id
chunkHmac = models.CharField(max_length=50, blank=False, null=False) return Path('user_{}'.format(user_id), col.uid, instance.uid)
class Meta:
unique_together = ('item', 'current')
def __str__(self):
return "{}, current={}".format(self.item.uid, self.current)
class CollectionItemChunk(models.Model): class CollectionItemChunk(models.Model):
uid = models.CharField(db_index=True, blank=False, null=False, uid = models.CharField(db_index=True, blank=False, null=False,
max_length=44, validators=[UidValidator]) max_length=44, validators=[UidValidator])
itemSnapshot = models.ForeignKey(CollectionItemSnapshot, related_name='chunks', null=True, on_delete=models.SET_NULL) items = models.ManyToManyField(CollectionItem, related_name='chunks')
order = models.CharField(max_length=100, blank=False, null=False) order = models.CharField(max_length=100, blank=False, null=False)
# We probably just want to implement this manually because we can have more than one pointing to a file. chunkFile = models.FileField(upload_to=chunk_directory_path)
class Meta: class Meta:
# unique_together = ('itemSnapshot', 'order') # Currently off because we set the item snapshot to null on deletion
ordering = ['order'] ordering = ['order']
def __str__(self): def __str__(self):

@ -64,25 +64,27 @@ class CollectionItemChunkSerializer(serializers.ModelSerializer):
fields = ('uid', ) fields = ('uid', )
class CollectionItemSnapshotSerializer(serializers.ModelSerializer): class CollectionItemSerializer(serializers.ModelSerializer):
encryptionKey = BinaryBase64Field()
chunks = serializers.SlugRelatedField( chunks = serializers.SlugRelatedField(
slug_field='uid', slug_field='uid',
queryset=models.CollectionItemChunk, queryset=models.CollectionItemChunk.objects.all(),
many=True many=True
) )
class Meta: class Meta:
model = models.CollectionItemSnapshot model = models.CollectionItem
fields = ('chunks', 'chunkHmac') fields = ('uid', 'version', 'encryptionKey', 'chunks', 'hmac')
class CollectionItemSerializer(serializers.ModelSerializer): class CollectionItemInlineSerializer(CollectionItemSerializer):
encryptionKey = BinaryBase64Field() chunksData = serializers.SerializerMethodField('get_inline_chunks_from_context')
content = CollectionItemSnapshotSerializer(
read_only=True,
many=False
)
class Meta: class Meta(CollectionItemSerializer.Meta):
model = models.CollectionItem fields = CollectionItemSerializer.Meta.fields + ('chunksData', )
fields = ('uid', 'version', 'encryptionKey', 'content')
def get_inline_chunks_from_context(self, obj):
request = self.context.get('request', None)
if request is not None:
return ['SomeInlineData', 'Somemoredata']
return 'readOnly'

@ -27,11 +27,11 @@ from rest_framework import viewsets
from rest_framework.response import Response from rest_framework.response import Response
from . import app_settings, paginators from . import app_settings, paginators
from .models import Collection, CollectionItem, CollectionItemSnapshot, CollectionItemChunk from .models import Collection, CollectionItem, CollectionItemChunk
from .serializers import ( from .serializers import (
CollectionSerializer, CollectionSerializer,
CollectionItemSerializer, CollectionItemSerializer,
CollectionItemSnapshotSerializer, CollectionItemInlineSerializer,
CollectionItemChunkSerializer CollectionItemChunkSerializer
) )

Loading…
Cancel
Save