Youngsun Lim commited on
Commit
901cd2b
ยท
1 Parent(s): a8fef31

app.py reward code

Browse files
Files changed (1) hide show
  1. app.py +17 -9
app.py CHANGED
@@ -247,18 +247,26 @@ def _persist_reward_code(pid: str, code: str, total_done: int):
247
  commit_message="append reward code"
248
  )
249
 
250
- def _gen_reward_code(pid: str, length: int = 10, forbid_ambiguous: bool=True) -> str:
251
  """
252
- ์ฐธ๊ฐ€์ž์—๊ฒŒ ๋ณด์—ฌ์ค„ ๋žœ๋ค ์ฝ”๋“œ. ์ถฉ๋Œ ์œ„ํ—˜ ๋งค์šฐ ๋‚ฎ์Œ.
253
- - PID + ์‹œ๊ฐ + ๋ณด์•ˆ๋žœ๋ค์œผ๋กœ ์‹œ๋“œ
254
- - ๋Œ€๋ฌธ์ž/์ˆซ์ž ์กฐํ•ฉ
255
  """
256
- alphabet = "ABCDEFGHJKLMNPQRSTUVWXYZ23456789" if forbid_ambiguous else string.ascii_uppercase + string.digits
257
- # secrets ๊ธฐ๋ฐ˜ ๋‚œ์ˆ˜ + pid/time ์„ž์–ด์„œ ํ•ด์‹œ ๋น„์Šทํ•œ ํšจ๊ณผ
258
- rnd = secrets.token_hex(8) + pid + str(time.time_ns())
259
- # ์„ž๊ณ  ๋ฝ‘๊ธฐ
 
 
 
 
 
 
260
  rng = secrets.SystemRandom()
261
- return "".join(rng.choice(alphabet) for _ in range(length))
 
 
262
 
263
 
264
 
 
247
  commit_message="append reward code"
248
  )
249
 
250
+ def _gen_reward_code(pid: str, length: int = 10, forbid_ambiguous: bool = True) -> str:
251
  """
252
+ ์ฐธ๊ฐ€์ž์—๊ฒŒ ๋ณด์—ฌ์ค„ ๋žœ๋ค ์ฝ”๋“œ.
253
+ - ํ•ญ์ƒ '2026'์œผ๋กœ ์‹œ์ž‘
254
+ - ๋’ค์— ๋Œ€๋ฌธ์ž/์ˆซ์ž ์กฐํ•ฉ ๋žœ๋ค ๋ฌธ์ž์—ด์„ ๋ถ™์—ฌ ์ด length ์ž๋ฆฟ์ˆ˜๋กœ ๋งŒ๋“ฆ
255
  """
256
+ prefix = "2026"
257
+ # prefix๋ฅผ ํฌํ•จํ•ด์„œ ์ „์ฒด ๊ธธ์ด๊ฐ€ length๊ฐ€ ๋˜๋„๋ก ๋’ค์— ๋ถ™์ผ ๊ธธ์ด ๊ณ„์‚ฐ
258
+ random_len = max(0, length - len(prefix))
259
+
260
+ if forbid_ambiguous:
261
+ # I, O, 0, 1 ์ œ๊ฑฐ
262
+ alphabet = "ABCDEFGHJKLMNPQRSTUVWXYZ23456789"
263
+ else:
264
+ alphabet = string.ascii_uppercase + string.digits
265
+
266
  rng = secrets.SystemRandom()
267
+ suffix = "".join(rng.choice(alphabet) for _ in range(random_len))
268
+
269
+ return prefix + suffix
270
 
271
 
272