116 lines
3.7 KiB
Python
116 lines
3.7 KiB
Python
from django.db import models
|
|
from . import available_themes, default_theme
|
|
from PIL import Image
|
|
|
|
|
|
def GetSettings():
|
|
s = Setting.objects.filter(id=1).first()
|
|
if s is None:
|
|
s = Setting(id=1)
|
|
s.save()
|
|
return s
|
|
|
|
|
|
def SettingExist(data):
|
|
s = GetSettings()
|
|
if data['page_theme'] == s.page_theme and data['page_title'] == s.page_title and data['page_image'] == s.page_image.name:
|
|
if s.page_image.name:
|
|
return data['page_image_data'] == s.page_image.open().read().hex()
|
|
else:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
|
|
class Setting(models.Model):
|
|
page_image = models.ImageField(
|
|
upload_to='theme',
|
|
default='theme/logo.png'
|
|
)
|
|
page_theme = models.CharField(
|
|
max_length=64,
|
|
choices=available_themes,
|
|
default=default_theme
|
|
)
|
|
page_title = models.CharField(max_length=32, default='MyApp', blank=True)
|
|
|
|
def save(self, *args, **kwargs):
|
|
super().save(*args, **kwargs)
|
|
|
|
img = Image.open(self.page_image.path)
|
|
if img.height > 50:
|
|
output_size = (int(img.width * 50 / img.height), 50)
|
|
img.thumbnail(output_size)
|
|
img.save(self.page_image.path)
|
|
|
|
def export_data(self):
|
|
rv = {}
|
|
rv['page_image'] = self.page_image.name
|
|
rv['page_image_data'] = self.page_image.open().read().hex()
|
|
rv['page_theme'] = self.page_theme
|
|
rv['page_title'] = self.page_title
|
|
return rv
|
|
|
|
def import_data(self, data):
|
|
self.page_title = data['page_title']
|
|
self.page_theme = data['page_theme']
|
|
self.page_image.name = data['page_image']
|
|
if self.page_image.name:
|
|
with self.page_image.open('wb') as pi:
|
|
pi.write(bytes.fromhex(data['page_image_data']))
|
|
self.save()
|
|
|
|
|
|
def BottomBarExist(data):
|
|
name = data.get('name', '')
|
|
icon_name = data.get('icon_name', '')
|
|
icon_data = data.get('icon_data')
|
|
url = data.get('url', '')
|
|
left = data.get('left')
|
|
sequence_number = data.get('sequence_number')
|
|
for bb_entry in BottomBar.objects.filter(name=name, icon=icon_name, url=url, left=left, sequence_number=sequence_number):
|
|
if icon_name == '' or icon_data == bb_entry.icon.open().read().hex():
|
|
return True
|
|
return False
|
|
|
|
|
|
class BottomBar(models.Model):
|
|
name = models.CharField(max_length=64, blank=True)
|
|
icon = models.ImageField(upload_to='theme', blank=True)
|
|
url = models.URLField()
|
|
left = models.BooleanField()
|
|
sequence_number = models.SmallIntegerField(choices=[(1, '1'), (2, '2'), (3, '3'), (4, '4'), (5, '5')])
|
|
|
|
def save(self, *args, **kwargs):
|
|
super().save(*args, **kwargs)
|
|
if self.icon:
|
|
img = Image.open(self.icon.path)
|
|
if img.height > 40:
|
|
output_size = (int(img.width * 40 / img.height), 40)
|
|
img.thumbnail(output_size)
|
|
img.save(self.icon.path)
|
|
|
|
def export_data(self):
|
|
rv = {}
|
|
rv['name'] = self.name
|
|
rv['icon_name'] = self.icon.name
|
|
if self.icon.name != '':
|
|
rv['icon_data'] = self.icon.open().read().hex()
|
|
else:
|
|
rv['icon_data'] = None
|
|
rv['url'] = self.url
|
|
rv['left'] = self.left
|
|
rv['sequence_number'] = self.sequence_number
|
|
return rv
|
|
|
|
def import_data(self, data):
|
|
self.name = data['name']
|
|
self.icon.name = data['icon_name']
|
|
if self.icon.name:
|
|
with self.icon.open('wb') as ih:
|
|
ih.write(bytes.fromhex(data['icon_data']))
|
|
self.url = data['url']
|
|
self.left = data['left']
|
|
self.sequence_number = data['sequence_number']
|
|
self.save()
|