CODEWITHSUNDEEP
python-3.xpostgresqlsqlalchemy
dada
dada

Reputation: 1493

Use a for loop to set columns of a table

I currently define the tables and columns of a PostgreSQL database like this:

from sqlalchemy.orm import declarative_base
from sqlalchemy import Column, Float, Integer
Base = declarative_base()
class MyTable(Base):
    __tablename__ = "my_table"
    col_1 = Column(Integer, primary_key=True)
    col_2 = Column(Float)
    col_3 = Column(Float)
    col_4 = Column(Float)
    col_5 = Column(Float)
    # ....
    col_24 = Column(Float)
    col_25 = Column(Float)

Instead of writing the same identical lines multiple times, I would like to use a for loop that sets similar columns:

from sqlalchemy.orm import declarative_base
from sqlalchemy import Column, Float, Integer
Base = declarative_base()
class MyTable(Base):
    __tablename__ = "my_table"
    col_1 = Column(Integer, primary_key=True)
    for ind in range(2, 26):
        # set col_$ind
        ????

What should I do put inside the for loop to set the attributes col_$ind ?

Upvotes: 1

Views: 840

Answers (2)

dada
dada

Reputation: 1493

I ended up using vars():

from sqlalchemy.orm import declarative_base
from sqlalchemy import Column, Float, Integer
Base = declarative_base()
class MyTable(Base):
    __tablename__ = "my_table"
    col_1 = Column(Integer, primary_key=True)
    for ind in range(2, 26):
        # set col_$ind
        vars()[f"col_{ind}"] = Column(Float)

Upvotes: 0

Yaakov Bressler
Yaakov Bressler

Reputation: 12098

Totally reasonable aim – as a software developer, your goal is to be efficient with your code. However, as you interact more deeply with relational databases, you may find value in explicit (though redundant) definitions.

Notwithstanding, here is how you may achieve loop-wise table definition:

from sqlalchemy.orm import declarative_base
from sqlalchemy import Column, Float, Integer

Base = declarative_base()

# Define your class here
class MyTable(Base):
    __tablename__ = "my_table"
    
    # define your primary key
    col_1 = Column(Integer, primary_key=True, comment='primary key')


# Create columns with a loop
for i in range(2, 26):
    setattr(MyTable, f'col_{i}', Column(Float))

Upvotes: 1

Related Questions