jerrrycans commited on
Commit
8a93644
·
verified ·
1 Parent(s): 108d181

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -10,7 +10,7 @@ app = Flask(__name__)
10
 
11
  # Initialize SQLite database
12
  def init_db():
13
- conn = sqlite3.connect('files.db')
14
  c = conn.cursor()
15
  c.execute('''CREATE TABLE IF NOT EXISTS files
16
  (code TEXT PRIMARY KEY, content TEXT)''')
@@ -23,7 +23,7 @@ def generate_unique_code():
23
  chars = string.ascii_letters + string.digits
24
  while True:
25
  code = ''.join(random.choice(chars) for _ in range(6))
26
- conn = sqlite3.connect('files.db')
27
  c = conn.cursor()
28
  c.execute("SELECT code FROM files WHERE code = ?", (code,))
29
  if not c.fetchone():
@@ -55,7 +55,7 @@ def handle_request():
55
 
56
  unique_code = generate_unique_code()
57
 
58
- conn = sqlite3.connect('files.db')
59
  c = conn.cursor()
60
  c.execute("INSERT INTO files (code, content) VALUES (?, ?)", (unique_code, content))
61
  conn.commit()
@@ -192,7 +192,7 @@ def view_raw_file(code):
192
  return view_file_internal(code, True)
193
 
194
  def view_file_internal(code, is_raw):
195
- conn = sqlite3.connect('files.db')
196
  c = conn.cursor()
197
  c.execute("SELECT content FROM files WHERE code = ?", (code,))
198
  result = c.fetchone()
 
10
 
11
  # Initialize SQLite database
12
  def init_db():
13
+ conn = sqlite3.connect('/tmp/files.db')
14
  c = conn.cursor()
15
  c.execute('''CREATE TABLE IF NOT EXISTS files
16
  (code TEXT PRIMARY KEY, content TEXT)''')
 
23
  chars = string.ascii_letters + string.digits
24
  while True:
25
  code = ''.join(random.choice(chars) for _ in range(6))
26
+ conn = sqlite3.connect('/tmp/files.db')
27
  c = conn.cursor()
28
  c.execute("SELECT code FROM files WHERE code = ?", (code,))
29
  if not c.fetchone():
 
55
 
56
  unique_code = generate_unique_code()
57
 
58
+ conn = sqlite3.connect('/tmp/files.db')
59
  c = conn.cursor()
60
  c.execute("INSERT INTO files (code, content) VALUES (?, ?)", (unique_code, content))
61
  conn.commit()
 
192
  return view_file_internal(code, True)
193
 
194
  def view_file_internal(code, is_raw):
195
+ conn = sqlite3.connect('/tmp/files.db')
196
  c = conn.cursor()
197
  c.execute("SELECT content FROM files WHERE code = ?", (code,))
198
  result = c.fetchone()