5035592165

Build Type

github

Committed by GitHub

Commit Message

Merge 9ce2b748a into 484aa0975

Pull Request Pull Request #378: More on references

Run Details

251 of 251 new or added lines in 30 files covered. (100.0%)

7464 of 8938 relevant lines covered (83.51%)

1.67 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

41.67

/winterdrp/utils/sql.py

"""
Util functions for database interactions
"""
from sqlalchemy import DDL, Engine, create_engine

from winterdrp.processors.database.postgres import DB_PASSWORD, DB_USER


def get_engine(
    db_user: str = DB_USER,
    db_password: str = DB_PASSWORD,
    db_name: str = "summer",
) -> Engine:
    """
    Function to create a postgres engine

    :param db_user: User for db
    :param db_password: password for db
    :param db_name: name of db
    :return: sqlalchemy engine
    """

    return create_engine(
        f"postgresql+psycopg://{db_user}:{db_password}" f"@/{db_name}",
        future=True,
    )


def create_q3c_extension(
    table_name: str, ra_column_name: str, dec_column_name: str, conn=None
):
    """
    Function to create q3c extension and index on table

    :param table_name: Name of table
    :param ra_column_name: ra column name
    :param dec_column_name: dec column name
    :param conn: connection to db
    :return:
    """

    trig_ddl = DDL(
        "CREATE EXTENSION IF NOT EXISTS q3c;"
        f"CREATE INDEX ON {table_name} "
        f"(q3c_ang2ipix({ra_column_name}, {dec_column_name}));"
        f"CLUSTER {table_name} USING {table_name}_q3c_ang2ipix_idx;"
        f"ANALYZE {table_name};"
    )

    if conn is None:
        with get_engine().connect() as conn:
            conn.execute(trig_ddl)
            conn.commit()
    else:
        conn.execute(trig_ddl)
        conn.commit()

1	"""
2	Util functions for database interactions
3	"""
4	from sqlalchemy import DDL, Engine, create_engine	2✔
5
6	from winterdrp.processors.database.postgres import DB_PASSWORD, DB_USER	2✔
7
8
9	def get_engine(	2✔
10	db_user: str = DB_USER,
11	db_password: str = DB_PASSWORD,
12	db_name: str = "summer",
13	) -> Engine:
14	"""
15	Function to create a postgres engine
16
17	:param db_user: User for db
18	:param db_password: password for db
19	:param db_name: name of db
20	:return: sqlalchemy engine
21	"""
22
23	return create_engine(	2✔
24	f"postgresql+psycopg://{db_user}:{db_password}" f"@/{db_name}",
25	future=True,
26	)
27
28
29	def create_q3c_extension(	2✔
30	table_name: str, ra_column_name: str, dec_column_name: str, conn=None
31	):
32	"""
33	Function to create q3c extension and index on table
34
35	:param table_name: Name of table
36	:param ra_column_name: ra column name
37	:param dec_column_name: dec column name
38	:param conn: connection to db
39	:return:
40	"""
41
42	trig_ddl = DDL(	×
43	"CREATE EXTENSION IF NOT EXISTS q3c;"
44	f"CREATE INDEX ON {table_name} "
45	f"(q3c_ang2ipix({ra_column_name}, {dec_column_name}));"
46	f"CLUSTER {table_name} USING {table_name}_q3c_ang2ipix_idx;"
47	f"ANALYZE {table_name};"
48	)
49
50	if conn is None:	×
51	with get_engine().connect() as conn:	×
52	conn.execute(trig_ddl)	×
53	conn.commit()	×
54	else:
55	conn.execute(trig_ddl)	×
56	conn.commit()	×

winter-telescope / winterdrp / 5035592165

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous