LSST Applications  21.0.0-147-g0e635eb1+1acddb5be5,22.0.0+052faf71bd,22.0.0+1ea9a8b2b2,22.0.0+6312710a6c,22.0.0+729191ecac,22.0.0+7589c3a021,22.0.0+9f079a9461,22.0.1-1-g7d6de66+b8044ec9de,22.0.1-1-g87000a6+536b1ee016,22.0.1-1-g8e32f31+6312710a6c,22.0.1-10-gd060f87+016f7cdc03,22.0.1-12-g9c3108e+df145f6f68,22.0.1-16-g314fa6d+c825727ab8,22.0.1-19-g93a5c75+d23f2fb6d8,22.0.1-19-gb93eaa13+aab3ef7709,22.0.1-2-g8ef0a89+b8044ec9de,22.0.1-2-g92698f7+9f079a9461,22.0.1-2-ga9b0f51+052faf71bd,22.0.1-2-gac51dbf+052faf71bd,22.0.1-2-gb66926d+6312710a6c,22.0.1-2-gcb770ba+09e3807989,22.0.1-20-g32debb5+b8044ec9de,22.0.1-23-gc2439a9a+fb0756638e,22.0.1-3-g496fd5d+09117f784f,22.0.1-3-g59f966b+1e6ba2c031,22.0.1-3-g849a1b8+f8b568069f,22.0.1-3-gaaec9c0+c5c846a8b1,22.0.1-32-g5ddfab5d3+60ce4897b0,22.0.1-4-g037fbe1+64e601228d,22.0.1-4-g8623105+b8044ec9de,22.0.1-5-g096abc9+d18c45d440,22.0.1-5-g15c806e+57f5c03693,22.0.1-7-gba73697+57f5c03693,master-g6e05de7fdc+c1283a92b8,master-g72cdda8301+729191ecac,w.2021.39
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: