75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
"""passbook app_gw models"""
|
|
import re
|
|
|
|
from django.contrib.postgres.fields import ArrayField
|
|
from django.db import models
|
|
from django.utils.translation import gettext as _
|
|
|
|
from passbook.core.models import Policy, PropertyMapping, Provider
|
|
|
|
|
|
class ApplicationGatewayProvider(Provider):
|
|
"""Virtual server which proxies requests to any hostname in server_name to upstream"""
|
|
|
|
server_name = ArrayField(models.TextField())
|
|
upstream = ArrayField(models.TextField())
|
|
enabled = models.BooleanField(default=True)
|
|
|
|
authentication_header = models.TextField(default='X-Remote-User', blank=True)
|
|
default_content_type = models.TextField(default='application/octet-stream')
|
|
upstream_ssl_verification = models.BooleanField(default=True)
|
|
|
|
form = 'passbook.providers.app_gw.forms.ApplicationGatewayProviderForm'
|
|
|
|
@property
|
|
def name(self):
|
|
"""since this model has no name property, return a joined list of server_names as name"""
|
|
return ', '.join(self.server_name)
|
|
|
|
def __str__(self):
|
|
return "Application Gateway %s" % ', '.join(self.server_name)
|
|
|
|
class Meta:
|
|
|
|
verbose_name = _('Application Gateway Provider')
|
|
verbose_name_plural = _('Application Gateway Providers')
|
|
|
|
|
|
class RewriteRule(PropertyMapping):
|
|
"""Rewrite requests matching `match` with `replacement`, if all polcies in `conditions` apply"""
|
|
|
|
REDIRECT_INTERNAL = 'internal'
|
|
REDIRECT_PERMANENT = 301
|
|
REDIRECT_FOUND = 302
|
|
|
|
REDIRECTS = (
|
|
(REDIRECT_INTERNAL, _('Internal')),
|
|
(REDIRECT_PERMANENT, _('Moved Permanently')),
|
|
(REDIRECT_FOUND, _('Found')),
|
|
)
|
|
|
|
match = models.TextField()
|
|
halt = models.BooleanField(default=False)
|
|
conditions = models.ManyToManyField(Policy, blank=True)
|
|
replacement = models.TextField() # python formatted strings, use {match.1}
|
|
redirect = models.CharField(max_length=50, choices=REDIRECTS)
|
|
|
|
form = 'passbook.providers.app_gw.forms.RewriteRuleForm'
|
|
|
|
_matcher = None
|
|
|
|
@property
|
|
def compiled_matcher(self):
|
|
"""Cache the compiled regex in memory"""
|
|
if not self._matcher:
|
|
self._matcher = re.compile(self.match)
|
|
return self._matcher
|
|
|
|
def __str__(self):
|
|
return "Rewrite Rule %s" % self.name
|
|
|
|
class Meta:
|
|
|
|
verbose_name = _('Rewrite Rule')
|
|
verbose_name_plural = _('Rewrite Rules')
|