def _reset_mapping():
with closing(get_connection()) as conn:
try:
cur = conn.cursor()
cur.execute('DELETE FROM mapping')
conn.commit()
except (psycopg2.ProgrammingError, psycopg2.InternalError) as e:
raise DatabaseError(e)
def delete_from_mapping(team):
with closing(get_connection()) as conn:
try:
cur = conn.cursor()
cur.execute('DELETE FROM mapping where team = %s;', (team,))
conn.commit()
except (psycopg2.ProgrammingError, psycopg2.InternalError) as e:
raise DatabaseError(e)
def get_mappings():
sql = """
SELECT team, app FROM mapping;
"""
with closing(get_connection()) as conn:
try:
cur = conn.cursor()
cur.execute(sql)
return cur.fetchall()
except (psycopg2.ProgrammingError, psycopg2.InternalError) as e:
raise DatabaseError(e)
def add_team_with_token(team, token):
sql = """
INSERT INTO mapping (team, token) VALUES (%s, %s);"""
with closing(get_connection()) as conn:
try:
cur = conn.cursor()
cur.execute(sql, (team, token))
conn.commit()
except psycopg2.IntegrityError:
raise DatabaseError(f'mapping already exists for {team}')
except (psycopg2.ProgrammingError, psycopg2.InternalError) as e:
raise DatabaseError(e)
def get_team_app_heroku_by_slack(token):
sql = """
SELECT team, app, heroku
FROM mapping
WHERE token = %s;
"""
with closing(get_connection()) as conn:
try:
cur = conn.cursor()
cur.execute(sql, (token,))
return cur.fetchone()
except (psycopg2.ProgrammingError, psycopg2.InternalError) as e:
raise DatabaseError(e)
def set_heroku_token_for_team(team, token):
sql = """
UPDATE mapping
SET heroku = %s
WHERE team = %s;
"""
with closing(get_connection()) as conn:
try:
cur = conn.cursor()
cur.execute(sql, (token, team))
conn.commit()
except (psycopg2.ProgrammingError, psycopg2.InternalError) as e:
raise DatabaseError(e)
def get_app_and_slack_token_for_team(team):
sql = """
SELECT app, token
FROM mapping
WHERE team = %s;
"""
with closing(get_connection()) as conn:
try:
cur = conn.cursor()
cur.execute(sql, (team,))
return cur.fetchone()
except (psycopg2.ProgrammingError, psycopg2.InternalError) as e:
raise DatabaseError(e)
def get_heroku_refresh_token_for_team(team):
sql = """
SELECT heroku_refresh
FROM mapping
WHERE team = %s;
"""
with closing(get_connection()) as conn:
try:
cur = conn.cursor()
cur.execute(sql, (team,))
return cur.fetchone()[0]
except (psycopg2.ProgrammingError, psycopg2.InternalError) as e:
raise DatabaseError(e)
except (IndexError, TypeError):
return
def create_mapping_table():
sql = """
CREATE TABLE mapping (
team varchar UNIQUE,
app varchar DEFAULT '',
token varchar DEFAULT '',
heroku varchar DEFAULT '',
heroku_refresh varchar DEFAULT '',
);"""
with closing(get_connection()) as conn:
try:
cur = conn.cursor()
cur.execute(sql)
conn.commit()
except (psycopg2.ProgrammingError, psycopg2.InternalError) as e:
raise DatabaseError(e)
def team_exists(team):
sql = """
SELECT *
FROM mapping
WHERE team = %s;
"""
with closing(get_connection()) as conn:
try:
cur = conn.cursor()
cur.execute(sql, (team,))
cur.fetchone()[0]
return True
except (psycopg2.ProgrammingError, psycopg2.InternalError) as e:
raise DatabaseError(e)
except (IndexError, TypeError):
return False