from django.db import models
from django.contrib.auth.models import User


class UserModel(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE, related_name='token_of')
    refresh_token = models.CharField(max_length=30, null=True, blank=True)
    gmb_account_id = models.CharField(max_length=30, null=True, blank=True)

    def __str__(self):
        return self.user.username


class Location(models.Model):
    location_id = models.CharField(max_length=50, unique=True)
    location_name = models.CharField(max_length=50)
    website_url = models.URLField()
    display_name = models.CharField(max_length=50)
    owner = models.ForeignKey(User, on_delete=models.CASCADE, null=True)

    def __str__(self):
        return self.display_name