Rename table to domain_audit and simplify subroutine addAudit().

Signed-off-by: DL6ER <dl6er@dl6er.de>
This commit is contained in:
DL6ER 2019-07-07 21:21:56 +02:00
parent be3e198f9a
commit 8382f4d727
No known key found for this signature in database
GPG key ID: 00135ACBD90B28DD
3 changed files with 12 additions and 8 deletions

View file

@ -18,7 +18,7 @@ upgrade_gravityDB(){
if [[ "$version" == "1" ]]; then
# This migration script upgrades the gravity.db file by
# adding the domain_auditlist table
# adding the domain_audit table
sqlite3 "${database}" < "/etc/.pihole/advanced/Scripts/database_migration/gravity/1_to_2.sql"
version=2
@ -26,7 +26,7 @@ upgrade_gravityDB(){
if [ -e "${auditFile}" ]; then
echo -e " ${INFO} Migrating content of ${auditFile} into new database"
# database_table_from_file is defined in gravity.sh
database_table_from_file "domain_auditlist" "${auditFile}"
database_table_from_file "domain_audit" "${auditFile}"
fi
fi
}

View file

@ -1,4 +1,4 @@
CREATE TABLE domain_auditlist
CREATE TABLE domain_audit
(
id INTEGER PRIMARY KEY AUTOINCREMENT,
domain TEXT UNIQUE NOT NULL,

View file

@ -557,23 +557,27 @@ addAudit()
shift # skip "-a"
shift # skip "audit"
local domains validDomain
domains="('$(checkDomain "${1}")')"
shift # skip first domain, as it has already been added
domains=""
for domain in "$@"
do
# Insert only the domain here. The date_added field will be
# filled with its default value (date_added = current timestamp)
validDomain="$(checkDomain "${domain}")"
if [[ -n "${validDomain}" ]]; then
domains="${domains},('${domain}')"
# Put comma in between () when there is
# more than one domains to be added
if [[ -n "${domains}" ]]; then
domains="${domains},"
fi
domains="${domains}('${domain}')"
fi
done
sqlite3 "${gravityDBfile}" "INSERT INTO \"domain_auditlist\" (domain) VALUES ${domains};"
sqlite3 "${gravityDBfile}" "INSERT INTO \"domain_audit\" (domain) VALUES ${domains};"
}
clearAudit()
{
sqlite3 "${gravityDBfile}" "DELETE FROM \"domain_auditlist\";"
sqlite3 "${gravityDBfile}" "DELETE FROM \"domain_audit\";"
}
SetPrivacyLevel() {