Spaces:

euler314
/

Purchase_record

Sleeping

App Files Files Community

euler314 commited on Feb 13

Commit

57fc3fb

verified ·

1 Parent(s): 55b202d

Update app.py

Browse files

Files changed (1) hide show

app.py +121 -31

app.py CHANGED Viewed

@@ -22,25 +22,32 @@ if CREDS_JSON:
                 'https://www.googleapis.com/auth/drive.file']
     )
-def ensure_headers():
     try:
         sheets_service = build('sheets', 'v4', credentials=CREDENTIALS)
-        # Directly update first row with headers
         headers = [
             ['姓名 Name', '學號 Student ID', '購買商品 Product',
              '金額 Cost', '發票檔案 Receipt File', '時間 Timestamp', 'IP位址 IP Address']
         ]
-        # Update A1:G1 with headers
         sheets_service.spreadsheets().values().update(
             spreadsheetId=SPREADSHEET_ID,
-            range='Sheet1!A1:G1',
             valueInputOption='RAW',
             body={'values': headers}
         ).execute()
-        # Format headers to be bold and centered
         requests = [{
             'repeatCell': {
                 'range': {
@@ -67,9 +74,6 @@ def ensure_headers():
     except Exception as e:
         print(f"Error setting headers: {str(e)}")
-@app.before_first_request
-def initialize():
-    ensure_headers()
 HTML_TEMPLATE = '''
 <!DOCTYPE html>
@@ -80,7 +84,7 @@ HTML_TEMPLATE = '''
     <style>
         body {
             font-family: Arial, sans-serif;
-            max-width: 600px;
             margin: 20px auto;
             padding: 20px;
         }
@@ -92,12 +96,13 @@ HTML_TEMPLATE = '''
             margin-bottom: 8px;
             font-weight: bold;
         }
-        input {
             width: 100%;
             padding: 10px;
             border: 1px solid #ddd;
             border-radius: 4px;
             font-size: 16px;
         }
         button {
             background-color: #4CAF50;
@@ -107,7 +112,6 @@ HTML_TEMPLATE = '''
             border-radius: 4px;
             cursor: pointer;
             font-size: 16px;
-            width: 100%;
         }
         button:hover {
             background-color: #45a049;
@@ -126,44 +130,118 @@ HTML_TEMPLATE = '''
             background-color: #f2dede;
             color: #a94442;
         }
     </style>
 </head>
 <body>
     <h2>購買記錄表單 Purchase Record Form</h2>
     <form id="purchaseForm">
         <div class="form-group">
-            <label for="name">姓名 Name:</label>
-            <input type="text" id="name" name="name" required>
         </div>
-        <div class="form-group">
-            <label for="studentId">學號 Student ID:</label>
-            <input type="text" id="studentId" name="studentId" required>
-        </div>
         <div class="form-group">
             <label for="product">購買商品 Product:</label>
             <input type="text" id="product" name="product" required>
         </div>
-        <div class="form-group">
-            <label for="cost">金額 Cost:</label>
-            <input type="number" id="cost" name="cost" required>
-        </div>
         <div class="form-group">
             <label for="receipt">發票 Receipt:</label>
             <input type="file" id="receipt" name="receipt" accept="image/*,.pdf" required>
         </div>
         <button type="submit">提交 Submit</button>
     </form>
     <div id="status"></div>
     <script>
         document.getElementById('purchaseForm').addEventListener('submit', async (e) => {
             e.preventDefault();
             const status = document.getElementById('status');
             status.style.display = 'block';
             status.textContent = '提交中... Submitting...';
@@ -183,6 +261,9 @@ HTML_TEMPLATE = '''
                     status.className = 'success';
                     status.textContent = '提交成功！ ' + result.message;
                     e.target.reset();
                 } else {
                     status.className = 'error';
                     status.textContent = '錯誤：' + result.message;
@@ -199,22 +280,24 @@ HTML_TEMPLATE = '''
 @app.route('/')
 def index():
-    return render_template_string(HTML_TEMPLATE)
 @app.route('/submit', methods=['POST'])
 def submit():
     try:
-        name = request.form['name']
-        student_id = request.form['studentId']
         product = request.form['product']
-        cost = request.form['cost']
         file = request.files['receipt']
         # Upload file to Drive
         drive_service = build('drive', 'v3', credentials=CREDENTIALS)
         file_metadata = {
-            'name': f"{student_id}_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}_{file.filename}",
             'parents': [FOLDER_ID]
         }
@@ -232,18 +315,25 @@ def submit():
         file_url = uploaded_file.get('webViewLink', '')
         # Update Google Sheet
         sheets_service = build('sheets', 'v4', credentials=CREDENTIALS)
         timestamp = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
         ip_address = request.remote_addr
-        values = [[name, student_id, product, cost, file_url, timestamp, ip_address]]
         body = {'values': values}
         sheets_service.spreadsheets().values().append(
             spreadsheetId=SPREADSHEET_ID,
-            range='Sheet1!A:G',
             valueInputOption='USER_ENTERED',
             body=body
         ).execute()

                 'https://www.googleapis.com/auth/drive.file']
     )
+def get_sheet_names():
+    try:
+        service = build('sheets', 'v4', credentials=CREDENTIALS)
+        spreadsheet = service.spreadsheets().get(spreadsheetId=SPREADSHEET_ID).execute()
+        return [sheet['properties']['title'] for sheet in spreadsheet['sheets']]
+    except Exception as e:
+        print(f"Error getting sheet names: {str(e)}")
+        return ['Sheet1']
+def ensure_headers(sheet_name):
     try:
         sheets_service = build('sheets', 'v4', credentials=CREDENTIALS)
         headers = [
             ['姓名 Name', '學號 Student ID', '購買商品 Product',
              '金額 Cost', '發票檔案 Receipt File', '時間 Timestamp', 'IP位址 IP Address']
         ]
         sheets_service.spreadsheets().values().update(
             spreadsheetId=SPREADSHEET_ID,
+            range=f'{sheet_name}!A1:G1',
             valueInputOption='RAW',
             body={'values': headers}
         ).execute()
+        # Format headers
         requests = [{
             'repeatCell': {
                 'range': {
     except Exception as e:
         print(f"Error setting headers: {str(e)}")
 HTML_TEMPLATE = '''
 <!DOCTYPE html>
     <style>
         body {
             font-family: Arial, sans-serif;
+            max-width: 800px;
             margin: 20px auto;
             padding: 20px;
         }
             margin-bottom: 8px;
             font-weight: bold;
         }
+        input, select {
             width: 100%;
             padding: 10px;
             border: 1px solid #ddd;
             border-radius: 4px;
             font-size: 16px;
+            margin-bottom: 10px;
         }
         button {
             background-color: #4CAF50;
             border-radius: 4px;
             cursor: pointer;
             font-size: 16px;
         }
         button:hover {
             background-color: #45a049;
             background-color: #f2dede;
             color: #a94442;
         }
+        .person-entry {
+            border: 1px solid #ddd;
+            padding: 15px;
+            margin-bottom: 15px;
+            border-radius: 4px;
+        }
+        .remove-person {
+            background-color: #dc3545;
+            margin-top: 10px;
+        }
+        #add-person {
+            margin-bottom: 20px;
+            background-color: #17a2b8;
+        }
+        .total-cost {
+            font-size: 1.2em;
+            font-weight: bold;
+            margin: 20px 0;
+        }
     </style>
 </head>
 <body>
     <h2>購買記錄表單 Purchase Record Form</h2>
     <form id="purchaseForm">
         <div class="form-group">
+            <label for="sheetSelect">選擇工作表 Select Sheet:</label>
+            <select id="sheetSelect" name="sheetName" required>
+                {% for sheet in sheet_names %}
+                    <option value="{{ sheet }}">{{ sheet }}</option>
+                {% endfor %}
+            </select>
         </div>
         <div class="form-group">
             <label for="product">購買商品 Product:</label>
             <input type="text" id="product" name="product" required>
         </div>
         <div class="form-group">
             <label for="receipt">發票 Receipt:</label>
             <input type="file" id="receipt" name="receipt" accept="image/*,.pdf" required>
         </div>
+        <div id="people-container">
+            <!-- Person entries will be added here -->
+        </div>
+        <button type="button" id="add-person">新增人員 Add Person</button>
+        <div class="total-cost">
+            總金額 Total Cost: <span id="total">0</span>
+        </div>
         <button type="submit">提交 Submit</button>
     </form>
     <div id="status"></div>
     <script>
+        let personCounter = 0;
+        function createPersonEntry() {
+            const container = document.createElement('div');
+            container.className = 'person-entry';
+            container.innerHTML = `
+                <div class="form-group">
+                    <label>姓名 Name:</label>
+                    <input type="text" name="names[]" required>
+                </div>
+                <div class="form-group">
+                    <label>學號 Student ID:</label>
+                    <input type="text" name="studentIds[]" required>
+                </div>
+                <div class="form-group">
+                    <label>金額 Cost:</label>
+                    <input type="number" name="costs[]" required onchange="updateTotal()">
+                </div>
+                <button type="button" class="remove-person" onclick="removePerson(this)">移除 Remove</button>
+            `;
+            return container;
+        }
+        function updateTotal() {
+            const costs = Array.from(document.getElementsByName('costs[]'))
+                .map(input => Number(input.value) || 0);
+            const total = costs.reduce((sum, cost) => sum + cost, 0);
+            document.getElementById('total').textContent = total;
+        }
+        function removePerson(button) {
+            button.parentElement.remove();
+            updateTotal();
+        }
+        document.getElementById('add-person').addEventListener('click', () => {
+            const container = document.getElementById('people-container');
+            container.appendChild(createPersonEntry());
+            personCounter++;
+        });
+        // Add initial person entry
+        document.getElementById('add-person').click();
         document.getElementById('purchaseForm').addEventListener('submit', async (e) => {
             e.preventDefault();
+            if (document.getElementsByName('names[]').length === 0) {
+                alert('請至少新增一個人員 Please add at least one person');
+                return;
+            }
             const status = document.getElementById('status');
             status.style.display = 'block';
             status.textContent = '提交中... Submitting...';
                     status.className = 'success';
                     status.textContent = '提交成功！ ' + result.message;
                     e.target.reset();
+                    document.getElementById('people-container').innerHTML = '';
+                    document.getElementById('add-person').click();
+                    document.getElementById('total').textContent = '0';
                 } else {
                     status.className = 'error';
                     status.textContent = '錯誤：' + result.message;
 @app.route('/')
 def index():
+    sheet_names = get_sheet_names()
+    return render_template_string(HTML_TEMPLATE, sheet_names=sheet_names)
 @app.route('/submit', methods=['POST'])
 def submit():
     try:
+        sheet_name = request.form['sheetName']
         product = request.form['product']
         file = request.files['receipt']
+        names = request.form.getlist('names[]')
+        student_ids = request.form.getlist('studentIds[]')
+        costs = request.form.getlist('costs[]')
         # Upload file to Drive
         drive_service = build('drive', 'v3', credentials=CREDENTIALS)
         file_metadata = {
+            'name': f"{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}_{file.filename}",
             'parents': [FOLDER_ID]
         }
         file_url = uploaded_file.get('webViewLink', '')
+        # Ensure headers exist
+        ensure_headers(sheet_name)
         # Update Google Sheet
         sheets_service = build('sheets', 'v4', credentials=CREDENTIALS)
         timestamp = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
         ip_address = request.remote_addr
+        # Create entries for each person
+        values = []
+        for name, student_id, cost in zip(names, student_ids, costs):
+            values.append([name, student_id, product, cost, file_url, timestamp, ip_address])
         body = {'values': values}
         sheets_service.spreadsheets().values().append(
             spreadsheetId=SPREADSHEET_ID,
+            range=f'{sheet_name}!A:G',
             valueInputOption='USER_ENTERED',
             body=body
         ).execute()