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


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

    class Meta:
        icon = 'media'



# 'Web based training Block'
class WebBasedTrainingBlock(blocks.StructBlock):
    RISE = 'rise'

    WBT_TYPE_CHOICES = (
        (RISE, 'Rise'),
    )

    url = blocks.URLBlock()
    type = models.CharField(
        max_length=100,
        choices=WBT_TYPE_CHOICES,
        default=RISE
    )

    class Meta:
        icon = 'media'

# 'Transver Task'
class TranverTaskBlock(blocks.StructBlock):
    title = models.CharField(max_length=128, default="")
    description = models.TextField(default="")

    class Meta:
        icon = 'media'