86 lines
2.5 KiB
Python
86 lines
2.5 KiB
Python
from django.db import models
|
|
|
|
|
|
class InvalidData(Exception):
|
|
''' Thrown when the DB contains invalid data, and cannot perform
|
|
something '''
|
|
|
|
def __init__(self, what):
|
|
self.what = what
|
|
super(InvalidData, self).__init__()
|
|
|
|
def __str__(self):
|
|
return self.what
|
|
|
|
|
|
class Keyword(models.Model):
|
|
''' A keyword/search term '''
|
|
|
|
text = models.CharField(max_length=256)
|
|
|
|
def __str__(self):
|
|
return self.text
|
|
|
|
|
|
class Webpage(models.Model):
|
|
url = models.URLField()
|
|
|
|
def __str__(self):
|
|
return self.url
|
|
|
|
|
|
class Website(models.Model):
|
|
''' A website usually visited '''
|
|
|
|
name = models.CharField(max_length=256)
|
|
url = models.URLField()
|
|
keywords = models.ManyToManyField(Keyword)
|
|
notable_pages = models.ManyToManyField(Webpage)
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
|
|
class BrowserFingerprint(models.Model):
|
|
''' A browser fingerprint, containing things like a user agent '''
|
|
|
|
description = models.CharField(max_length=256)
|
|
useragent = models.CharField(max_length=256)
|
|
appname = models.CharField(max_length=256)
|
|
appversion = models.CharField(max_length=256)
|
|
platform = models.CharField(max_length=256)
|
|
vendor = models.CharField(max_length=256)
|
|
vendorsub = models.CharField(max_length=256)
|
|
buildID = models.CharField(max_length=256)
|
|
oscpu = models.CharField(max_length=256)
|
|
accept_encoding = models.CharField(max_length=256)
|
|
accept_default = models.CharField(max_length=256)
|
|
accept_lang = models.CharField(max_length=256)
|
|
pixeldepth = models.IntegerField()
|
|
colordepth = models.IntegerField()
|
|
screens = models.CharField(max_length=256)
|
|
|
|
def __str__(self):
|
|
return self.description
|
|
|
|
|
|
class SearchEngine(models.Model):
|
|
''' A search engine, and all the data needed to use it '''
|
|
|
|
name = models.CharField(max_length=256)
|
|
url = models.URLField()
|
|
query_pattern = models.CharField(max_length=256) # This field is the
|
|
# query pattern. It should contain a `{}`, which, when substituted with a
|
|
# search term (using `.format()`), must yield a URL that can be resolved to
|
|
# perform the search
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def search_url(self, search_term):
|
|
''' Obtain a url to search `search_term` with this search engine '''
|
|
pattern = str(self.query_pattern)
|
|
if '{}' not in pattern:
|
|
raise InvalidData("Search engine {}: bad pattern".format(self))
|
|
return str(self.query_pattern).format(search_term)
|