파이썬, 연속된 숫자 ID 생성 함수 예제

파이썬, 연속된 숫자 ID 생성 함수 예제 

 

글. 수알치 오상문

 

 

연속된 숫자를 생성하는 제너레이터 함수 예제입니다. 

생성 함수를 변형하면 다른 유형의 생성도 가능합니다. 

 

class IdGenerator:
  def __init__(self, last_id):
    self.user_last_id = last_id

  def get_new_id(self):
    self.user_last_id += 1
    return self.user_last_id

# 테스트 -------------------------------
users = [{'id':1, 'name':'tom'},
         {'id':2, 'name':'jobs'}]

id_gen = IdGenerator(len(users))
id_make = id_gen.get_new_id

users.append({'id':id_make(), 'name':'sony'})
users.append({'id':id_make(), 'name':'sunny'})

print(users)
print("-"*80)
print(*users, sep="\n")

 

[실행 결과]

[{'id': 1, 'name': 'tom'}, {'id': 2, 'name': 'jobs'}, {'id': 3, 'name': 'sony'}, {'id': 4, 'name': 'sunny'}]
--------------------------------------------------------------------------------
{'id': 1, 'name': 'tom'}
{'id': 2, 'name': 'jobs'}
{'id': 3, 'name': 'sony'}
{'id': 4, 'name': 'sunny'}