2024-07-22 Backend Code Review

parasti šo izmantojam psycopg2

asyncpg https://www.perplexity.ai/search/asyncpg-named-sql-query-mdyjT5JGQP2b48svzS6xfg

Piemērs kā jāizskatās ControllerDatabase funkcijām

xxxxxxxxxx
2
1
def get_tasks_by_feature(self, feature: DbFeature = None, language_code: LanguageCodes = None) -> List[DbFeaturesInTask]:
2
    results = []

Startup uztaisa pool un tad ControllerDatabase izmanto https://www.perplexity.ai/search/give-example-fastapi-psycopg2-C.04CzVHSYadcs8xowkCZQ

Select ar zvaigznīti

xxxxxxxxxx
13
1
      with self.db_context() as cursor:
2
        cursor.execute("SELECT * FROM features_workers")
3

4
        if cursor.rowcount > 0:
5
          for row in cursor.fetchall():
6
            columns = [it[0] for it in cursor.description]
7
            feature_worker_dictionary = dict(zip(columns, row))
8
            feature_worker: DbFeatureWorker = from_dict(
9
              data_class=DbFeatureWorker,
10
              data=feature_worker_dictionary,
11
              config=Config(check_types=False)
12
            )
13
            feature_workers.append(feature_worker)

​x
1
  def convert_enum_to_str(self, data_dict):
2
    for key, value in data_dict.items():
3
      if isinstance(value, Enum) or isinstance(value, datetime.datetime):
4
        data_dict[key] = str(value)
5
      elif isinstance(value, dict):
6
        data_dict[key] = self.convert_enum_to_str(data_dict[key])
7
      elif isinstance(value, list):
8
        data_dict[key] = self.convert_list_enum_to_str(data_dict[key])
9
    return data_dict
10

11
  def convert_list_enum_to_str(self, data_list):
12
    # concatenate LanguageCodes
13
    if all([type(i) is LanguageCodes for i in data_list]):
14
      return ','.join([str(i) for i in data_list])
15
    for idx in range(len(data_list)):
16
      value = data_list[idx]
17
      if isinstance(value, Enum) or isinstance(value, datetime.datetime):
18
        data_list[idx] = str(value)
19
      elif isinstance(value, dict):
20
        data_list[idx] = self.convert_enum_to_str(data_list[idx])
21
      elif isinstance(value, list):
22
        data_list[idx] = self.convert_list_enum_to_str(data_list[idx])
23
    return data_list
24

25
  def convert_numpy_to_bytes(self, data_dict):
26
    for key, value in data_dict.items():
27
      if isinstance(value, np.ndarray):
28
        data_dict[key] = value.astype(np.float32).tobytes()
29
    return data_dict
30