Updated banners.py

Removed unnecessary import.
Added a function to avoid duplicating code.
This commit is contained in:
boreq 2014-09-24 01:34:31 +02:00
parent 9e009b987c
commit 78f1a99820

View File

@ -1,31 +1,31 @@
<?php <?php
include 'inc/functions.php';
header("Pragma-directive: no-cache"); header("Pragma-directive: no-cache");
header("Cache-directive: no-cache"); header("Cache-directive: no-cache");
header("Cache-control: no-cache"); header("Cache-control: no-cache");
header("Pragma: no-cache"); header("Pragma: no-cache");
header("Expires: 0"); header("Expires: 0");
if (!isset($_GET['board'])) {
header("Location: static/8chan banner.png");
die();
}
$b = $_GET['board'];
if (!preg_match('/^[a-z0-9]{1,10}$/', $b))
header("Location: static/8chan banner.png");
$dir = 'static/banners/'.$b; function get_custom_banner(&$b) {
# Validate the board name
if (!(isset($b) && preg_match('/^[a-z0-9]{1,10}$/', $b)))
return null;
if (is_dir($dir)) { # Check if directory exists
$banners = array_diff(scandir($dir), array('..', '.')); $dir = "static/banners/$b/";
$r = array_rand($banners); if (!is_dir($dir))
} else { return null;
$banners = array();
# Return random file if directory is not empty
$banners = array_diff(scandir($dir), array('..', '.'));
if (!$banners)
return null;
$r = array_rand($banners);
return $dir.$banners[$r];
} }
if (!empty($banners)) { $banner = get_custom_banner($_GET['board']);
header("Location: $dir/{$banners[$r]}"); if ($banner)
} else { header("Location: $banner");
header("Location: static/8chan banner.png"); else
} header("Location: static/8chan banner.png");