__init__.py

Go to the documentation of this file.

import os.path
import sqlite3
 
import Crossfire
 
def _init_schema(con, version, *schema_files):
    con.execute("PRAGMA journal_mode=WAL;");
    con.execute("PRAGMA synchronous=NORMAL;");
    con.execute("CREATE TABLE IF NOT EXISTS schema(version INT);");
    # We will always return something, either a zero or the highest version number.
    result = con.execute("SELECT COALESCE(MAX(version), 0) FROM schema").fetchall();
    curr = result[0][0]
    if curr < version:
        Crossfire.Log(Crossfire.LogInfo,
                "Initializing factions schema %d->%d" % (curr, version))
        if curr == 0:
            # schema.sql is already updated to load in the current schema.
            with open(schema_files[0]) as initfile:
                con.executescript(initfile.read())
        else:
            for f in schema_files:
                # Compare just the file name
                if f.split("/").pop() == f"update_schema_{curr}_{curr+1}.sql":
                    with open(f) as updfile:
                        con.executescript(updfile.read())
                    curr += 1
        con.commit()
 
def _get_sql_path(f):
    return os.path.join(Crossfire.DataDirectory(), Crossfire.MapDirectory(),
            "python/CFReputation/sql", f)
 
def _init_db():
    # Schema update files must go in order.
    schema_files = map(_get_sql_path, ["schema.sql", "update_schema_1_2.sql"])
    init_files = map(_get_sql_path, ["init.sql", "gods.sql"])
    db_path = os.path.join(Crossfire.LocalDirectory(), "factions.db")
    con = sqlite3.connect(db_path)
    _init_schema(con, 2, *schema_files)
    for f in init_files:
        with open(f) as initfile:
            con.executescript(initfile.read())
    return con
 
def _get_db():
    return _init_db()
 
def reputation(player, faction=None):
    """
    Return tuple with the name and reputation of the player with the given
    faction. If faction is None, return all known reputations.
    """
    con = _get_db()
    if faction is None:
        query="""
SELECT faction, CAST(ROUND(reputation*100) as integer) as rep
FROM reputations
WHERE name=? AND ABS(rep) > 0;
        """
        result = con.execute(query, (player,)).fetchall()
    else:
        query="""
SELECT faction, CAST(ROUND(reputation*100) as integer) as rep
FROM reputations
WHERE name=? AND faction=? AND ABS(rep) > 0;
        """
        result = con.execute(query, (player, faction)).fetchall()
    con.close()
    return result
 
def record_kill(race, region, player, fraction=0.0001, limit=0.4):
    con = _get_db()
    query = """
WITH updates AS (
    SELECT faction, -attitude*? AS change
    FROM regions
    NATURAL JOIN relations
    WHERE race=? AND (region=? OR region='ALL'))
REPLACE INTO reputations
SELECT ? AS player, updates.faction,
    COALESCE(reputation, 0) + change AS new_rep
FROM updates
LEFT JOIN reputations
    ON updates.faction=reputations.faction AND player=reputations.name
WHERE ABS(new_rep) <= ?;
    """
    con.execute(query, (fraction, race, region, player, limit))
    con.commit()
    con.close()