Linux-specific dotfiles

1 files changed, 153 insertions, 0 deletions

diff --git a/linux/home/.scripts/random_data.py b/linux/home/.scripts/random_data.py
new file mode 100755
index 0000000..071ab7c
--- /dev/null
+++ b/linux/home/.scripts/random_data.py
@@ -0,0 +1,153 @@
+#!/usr/bin/env python3
+
+import os
+import random
+import string
+import json
+import datetime
+import csv
+
+
+def generate_random_string(length, charset=string.ascii_letters):
+    """Generate a random string of given length and character set."""
+    return ''.join(random.choice(charset) for _ in range(length))
+
+
+def generate_random_number(min_value, max_value):
+    """Generate a random number within the specified range."""
+    return random.randint(min_value, max_value)
+
+
+def generate_random_date(start_date, end_date):
+    """Generate a random date within the specified range."""
+    time_between_dates = end_date - start_date
+    days_between_dates = time_between_dates.days
+    random_number_of_days = random.randrange(days_between_dates)
+    random_date = start_date + datetime.timedelta(days=random_number_of_days)
+    return random_date.strftime("%Y-%m-%d")
+
+
+def generate_sql_insert(table_name, columns, num_records):
+    """Generate SQL INSERT statements for populating a table."""
+    sql_statements = []
+    for _ in range(num_records):
+        values = [f"'{generate_random_string(int(input(f'Enter length for {column}: ')))}'" for column in columns]
+        sql = f"INSERT INTO {table_name} ({', '.join(columns)}) VALUES ({', '.join(values)});"
+        sql_statements.append(sql)
+    return sql_statements
+
+
+def generate_placeholder_data(num_records, data_format):
+    """Generate placeholder data based on user-defined format."""
+    placeholder_data = []
+    for _ in range(num_records):
+        record = {}
+        for field, field_data in data_format.items():
+            data_type = field_data['type']
+            if data_type == 'string':
+                length = field_data['length']
+                charset = field_data['charset']
+                record[field] = generate_random_string(length, charset)
+            elif data_type == 'number':
+                min_value = field_data['min']
+                max_value = field_data['max']
+                record[field] = generate_random_number(min_value, max_value)
+            elif data_type == 'date':
+                start_date = datetime.datetime.strptime(field_data['start_date'], "%Y-%m-%d")
+                end_date = datetime.datetime.strptime(field_data['end_date'], "%Y-%m-%d")
+                record[field] = generate_random_date(start_date, end_date)
+            elif data_type == 'boolean':
+                record[field] = random.choice([True, False])
+        placeholder_data.append(record)
+    return placeholder_data
+
+
+def get_data_format_from_user():
+    """Prompt the user for data format preferences."""
+    data_format = {}
+    while True:
+        field = input("Enter field name (or 'done' to finish): ").strip()
+        if field.lower() == 'done':
+            break
+
+        data_type = input(f"Enter data type for '{field}' (string/number/date/boolean): ").strip()
+        if data_type not in ['string', 'number', 'date', 'boolean']:
+            print("Invalid data type. Please enter 'string', 'number', 'date', or 'boolean'.")
+            continue
+
+        if data_type == 'string':
+            length = int(input(f"Enter length for '{field}' (integer): "))
+            charset = input(f"Enter character set for '{field}' (optional, press Enter for default): ").strip()
+            if not charset:
+                charset = string.ascii_letters
+            data_format[field] = {'type': 'string', 'length': length, 'charset': charset}
+        elif data_type == 'number':
+            min_value = int(input(f"Enter minimum value for '{field}' (integer): "))
+            max_value = int(input(f"Enter maximum value for '{field}' (integer): "))
+            data_format[field] = {'type': 'number', 'min': min_value, 'max': max_value}
+        elif data_type == 'date':
+            start_date = input(f"Enter start date for '{field}' (YYYY-MM-DD): ")
+            end_date = input(f"Enter end date for '{field}' (YYYY-MM-DD): ")
+            data_format[field] = {'type': 'date', 'start_date': start_date, 'end_date': end_date}
+        elif data_type == 'boolean':
+            data_format[field] = {'type': 'boolean'}
+
+    return data_format
+
+
+def get_file_type_from_user():
+    """Prompt the user for the desired file type (e.g., JSON, CSV, SQL, TXT, MD, HTML)."""
+    while True:
+        file_type = input("Enter the desired file type for saving the data (json/csv/sql/txt/md/html): ").strip().lower()
+        if file_type in ['json', 'csv', 'sql', 'txt', 'md', 'html']:
+            return file_type
+        else:
+            print("Invalid file type. Please enter 'json', 'csv', 'sql', 'txt', 'md', or 'html'.")
+
+
+def save_data_to_file(data, file_type):
+    """Save the generated data to a file of the specified type."""
+    if file_type == 'json':
+        with open('placeholder_data.json', 'w') as json_file:
+            json.dump(data, json_file, indent=4)
+    elif file_type == 'csv':
+        with open('placeholder_data.csv', 'w', newline='') as csv_file:
+            fieldnames = data[0].keys()
+            writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
+            writer.writeheader()
+            for record in data:
+                writer.writerow(record)
+    elif file_type == 'sql':
+        table_name = input("Enter the SQL table name: ")
+        columns = input("Enter column names separated by commas: ").split(',')
+        sql_statements = generate_sql_insert(table_name, columns, len(data))
+        with open('generated_data.sql', 'w') as sql_file:
+            sql_file.write('\n'.join(sql_statements))
+    elif file_type == 'txt':
+        with open('placeholder_data.txt', 'w') as txt_file:
+            for record in data:
+                txt_file.write(str(record) + '\n')
+    elif file_type == 'md':
+        with open('placeholder_data.md', 'w') as md_file:
+            for record in data:
+                md_file.write('- ' + ', '.join([f"{key}: {value}" for key, value in record.items()]) + '\n')
+    elif file_type == 'html':
+        with open('placeholder_data.html', 'w') as html_file:
+            html_file.write('<html>\n<head>\n<title>Placeholder Data</title>\n</head>\n<body>\n')
+            for record in data:
+                html_file.write('<ul>\n')
+                for key, value in record.items():
+                    html_file.write(f'<li>{key}: {value}</li>\n')
+                html_file.write('</ul>\n')
+
+
+if __name__ == "__main__":
+    num_records = int(input("Enter the number of records to generate: "))
+    data_format = get_data_format_from_user()
+    file_type = get_file_type_from_user()
+
+    placeholder_data = generate_placeholder_data(num_records, data_format)
+
+    save_data_to_file(placeholder_data, file_type)
+
+    print(f"Data will be saved to: {os.path.abspath('generated_data.sql')}")