LSST Applications  21.0.0-172-gfb10e10a+18fedfabac,22.0.0+297cba6710,22.0.0+80564b0ff1,22.0.0+8d77f4f51a,22.0.0+a28f4c53b1,22.0.0+dcf3732eb2,22.0.1-1-g7d6de66+2a20fdde0d,22.0.1-1-g8e32f31+297cba6710,22.0.1-1-geca5380+7fa3b7d9b6,22.0.1-12-g44dc1dc+2a20fdde0d,22.0.1-15-g6a90155+515f58c32b,22.0.1-16-g9282f48+790f5f2caa,22.0.1-2-g92698f7+dcf3732eb2,22.0.1-2-ga9b0f51+7fa3b7d9b6,22.0.1-2-gd1925c9+bf4f0e694f,22.0.1-24-g1ad7a390+a9625a72a8,22.0.1-25-g5bf6245+3ad8ecd50b,22.0.1-25-gb120d7b+8b5510f75f,22.0.1-27-g97737f7+2a20fdde0d,22.0.1-32-gf62ce7b1+aa4237961e,22.0.1-4-g0b3f228+2a20fdde0d,22.0.1-4-g243d05b+871c1b8305,22.0.1-4-g3a563be+32dcf1063f,22.0.1-4-g44f2e3d+9e4ab0f4fa,22.0.1-42-gca6935d93+ba5e5ca3eb,22.0.1-5-g15c806e+85460ae5f3,22.0.1-5-g58711c4+611d128589,22.0.1-5-g75bb458+99c117b92f,22.0.1-6-g1c63a23+7fa3b7d9b6,22.0.1-6-g50866e6+84ff5a128b,22.0.1-6-g8d3140d+720564cf76,22.0.1-6-gd805d02+cc5644f571,22.0.1-8-ge5750ce+85460ae5f3,master-g6e05de7fdc+babf819c66,master-g99da0e417a+8d77f4f51a,w.2021.48
LSST Data Management Base Package
Public Member Functions | Public Attributes | List of all members
lsst.pipe.tasks.ingestPgsql.PgsqlRegistryContext Class Reference
Inheritance diagram for lsst.pipe.tasks.ingestPgsql.PgsqlRegistryContext:
lsst.pipe.tasks.ingest.RegistryContext

Public Member Functions

def __init__ (self, registryName, createTableFunc, forceCreateTables)
 
def __exit__ (self, excType, excValue, traceback)
 
def __enter__ (self)
 

Public Attributes

 registryName
 
 conn
 

Detailed Description

Context manager to provide a pgsql registry

Definition at line 14 of file ingestPgsql.py.

Constructor & Destructor Documentation

◆ __init__()

def lsst.pipe.tasks.ingestPgsql.PgsqlRegistryContext.__init__ (   self,
  registryName,
  createTableFunc,
  forceCreateTables 
)
Construct a context manager

@param registryName: Name of registry file
@param createTableFunc: Function to create tables
@param forceCreateTables: Force the (re-)creation of tables?

Definition at line 17 of file ingestPgsql.py.

17  def __init__(self, registryName, createTableFunc, forceCreateTables):
18  """Construct a context manager
19 
20  @param registryName: Name of registry file
21  @param createTableFunc: Function to create tables
22  @param forceCreateTables: Force the (re-)creation of tables?
23  """
24  self.registryName = registryName
25  data = PgsqlRegistry.readYaml(registryName)
26  self.conn = pgsql.connect(host=data["host"], port=data["port"], user=data["user"],
27  password=data["password"], database=data["database"])
28  cur = self.conn.cursor()
29 
30  # Check for existence of tables
31  cur.execute("SELECT relname FROM pg_class WHERE relkind='r' AND relname='raw'")
32  rows = cur.fetchall()
33 
34  if forceCreateTables or len(rows) == 0:
35  # Delete all tables and start over.
36  # Not simply doing "DROP SCHEMA" and "CREATE SCHEMA" because of permissions.
37  cur.execute("SELECT tablename FROM pg_tables WHERE schemaname = 'public'")
38  tables = cur.fetchall()
39  for tt in tables:
40  cur.execute("DROP TABLE %s CASCADE" % tt)
41  createTableFunc(self.conn)
42 

Member Function Documentation

◆ __enter__()

def lsst.pipe.tasks.ingest.RegistryContext.__enter__ (   self)
inherited
Provide the 'as' value

Definition at line 261 of file ingest.py.

261  def __enter__(self):
262  """Provide the 'as' value"""
263  return self.conn
264 

◆ __exit__()

def lsst.pipe.tasks.ingestPgsql.PgsqlRegistryContext.__exit__ (   self,
  excType,
  excValue,
  traceback 
)

Reimplemented from lsst.pipe.tasks.ingest.RegistryContext.

Definition at line 43 of file ingestPgsql.py.

43  def __exit__(self, excType, excValue, traceback):
44  self.conn.commit()
45  self.conn.close()
46  return False # Don't suppress any exceptions
47 
48 

Member Data Documentation

◆ conn

lsst.pipe.tasks.ingestPgsql.PgsqlRegistryContext.conn

Definition at line 26 of file ingestPgsql.py.

◆ registryName

lsst.pipe.tasks.ingestPgsql.PgsqlRegistryContext.registryName

Definition at line 24 of file ingestPgsql.py.


The documentation for this class was generated from the following file: