from django.db import models
from wagtail.core import blocks


class VideoBlock(blocks.StructBlock):
    # TODO: Possible video Types for the user, upload file, add URL
    title = models.CharField(max_length=255, default="")
    description = models.TextField(default="")
    url = blocks.URLBlock()

    class Meta:
        icon = 'media'


RISE = 'rise'

CONTENT_TYPE_CHOICES = (
    (RISE, 'Rise'),
)


class WebBasedTrainingBlock(blocks.StructBlock):
    url = blocks.URLBlock()
    content_type = models.CharField(
        max_length=255,
        choices=CONTENT_TYPE_CHOICES,
        default=RISE
    )

    class Meta:
        icon = 'media'