Add main to scripts

Author: dltamayo

bin/compare_clonal_publicity.py

@@ -6,42 +6,49 @@
 Output: .tsv of TCR sharing across samples, .tsv of sample-sample_id mapping 
 """
 import argparse
-
 import pandas as pd
 
-# Initialize the parser
-parser = argparse.ArgumentParser(description="Take positional args")
-
-# Add positional arguments
-parser.add_argument("cdr_df")
-args = parser.parse_args()
-
-# Load your data
-df = pd.read_csv(args.cdr_df, sep="\t")
-
-# Step 1: Map samples to integers
-sample_mapping = {sample: i + 1 for i, sample in enumerate(df['sample'].unique())}
-df['sample_id'] = df['sample'].map(sample_mapping)
-
-# Step 2: Group by CDR3b and aggregate sample_ids
-grouped = (
-    df.groupby('CDR3b')['sample_id']
-    .apply(lambda x: sorted(set(x)))  # remove duplicates if any
-    .reset_index()
-)
-
-# Step 3: Add comma-separated list and total count
-grouped['samples_present'] = grouped['sample_id'].apply(lambda x: ",".join(map(str, x)))
-grouped['total_samples'] = grouped['sample_id'].apply(len)
-
-# Step 4: Final output — drop raw list
-final_df = grouped[['CDR3b', 'total_samples', 'samples_present']]
-final_df = final_df.sort_values(by='total_samples', axis=0, ascending=False)
-
-# Step 5: Export both outputs
-final_df.to_csv("cdr3_sharing.tsv", sep="\t", index=False)
-
-# Also export the sample mapping
-sample_map_df = pd.DataFrame.from_dict(sample_mapping, orient='index', columns=['sample_id']).reset_index()
-sample_map_df.columns = ['patient', 'sample_id']
-sample_map_df.to_csv("sample_mapping.tsv", sep="\t", index=False)
+def main():
+    # Initialize the parser
+    parser = argparse.ArgumentParser(description="Take positional args")
+    
+    # Add positional arguments
+    parser.add_argument(
+        "cdr_df",
+        type=str,
+        help="Input file name, expected to be in TSV format with columns [CDR3b, sample]",
+    )
+    args = parser.parse_args()
+    
+    # Load data
+    df = pd.read_csv(args.cdr_df, sep="\t")
+    
+    # Step 1: Map samples to integers
+    sample_mapping = {sample: i + 1 for i, sample in enumerate(df['sample'].unique())}
+    df['sample_id'] = df['sample'].map(sample_mapping)
+    
+    # Step 2: Group by CDR3b and aggregate sample_ids
+    grouped = (
+        df.groupby('CDR3b')['sample_id']
+        .apply(lambda x: sorted(set(x)))  # remove duplicates if any
+        .reset_index()
+    )
+    
+    # Step 3: Add comma-separated list and total count
+    grouped['samples_present'] = grouped['sample_id'].apply(lambda x: ",".join(map(str, x)))
+    grouped['total_samples'] = grouped['sample_id'].apply(len)
+    
+    # Step 4: Final output — drop raw list
+    final_df = grouped[['CDR3b', 'total_samples', 'samples_present']]
+    final_df = final_df.sort_values(by='total_samples', axis=0, ascending=False)
+    
+    # Step 5: Export both outputs
+    final_df.to_csv("cdr3_sharing.tsv", sep="\t", index=False)
+    
+    # Also export the sample mapping
+    sample_map_df = pd.DataFrame.from_dict(sample_mapping, orient='index', columns=['sample_id']).reset_index()
+    sample_map_df.columns = ['patient', 'sample_id']
+    sample_map_df.to_csv("sample_mapping.tsv", sep="\t", index=False)
+
+if __name__ == "__main__":
+    main()

bin/compare_concatenate.py

@@ -8,59 +8,62 @@
 
 # Import modules
 import argparse
-import glob
 import os
 import pandas as pd
 
-# Initialize the parser
-parser = argparse.ArgumentParser(description="Take positional args")
+def main():
+    # Initialize the parser
+    parser = argparse.ArgumentParser(description="Take positional args")
 
-# Add positional arguments
-parser.add_argument("data_dir")
-parser.add_argument("samplesheet")
+    # Add positional arguments
+    parser.add_argument("data_dir")
+    parser.add_argument("samplesheet")
 
-# Parse the arguments
-args = parser.parse_args()
+    # Parse the arguments
+    args = parser.parse_args()
 
-# Print the arguments
-print("data_dir: ", args.data_dir)
-print("samplesheet: ", args.samplesheet)
+    # Print the arguments
+    print("data_dir: ", args.data_dir)
+    print("samplesheet: ", args.samplesheet)
 
-samplesheet = pd.read_csv(args.samplesheet, header=0)
+    samplesheet = pd.read_csv(args.samplesheet, header=0)
 
-dfs = []
-for index, row in samplesheet.iterrows():
-    file_path = os.path.basename(row['file'])
-    file_path = os.path.join(args.data_dir, file_path)
-    print(f"Loading {file_path}")
-    
-    # Read the TSV file into a dataframe
-    df = pd.read_csv(file_path, sep="\t", header=0)
-    
-    # Get metadata
-    subject_id = row['subject_id']
-    timepoint = row['timepoint']
-    origin = row['origin']
+    dfs = []
+    for index, row in samplesheet.iterrows():
+        file_path = os.path.basename(row['file'])
+        file_path = os.path.join(args.data_dir, file_path)
+        print(f"Loading {file_path}")
 
-    # Add patient column
-    df['patient'] = f"{subject_id}:{timepoint}_{origin}"
-    df['sample'] = row['sample']
-    
-    # Select relevant columns
-    df = df[['aminoAcid', 'vGeneName', 'jGeneName', 'patient', 'count (templates/reads)', 'sample']]
-    dfs.append(df)
+        # Read the TSV file into a dataframe
+        df = pd.read_csv(file_path, sep="\t", header=0)
+        
+        # Get metadata
+        subject_id = row['subject_id']
+        timepoint = row['timepoint']
+        origin = row['origin']
+            
+        # Add patient column
+        df['patient'] = f"{subject_id}:{timepoint}_{origin}"
+        df['sample'] = row['sample']
+        
+        # Select relevant columns
+        df = df[['aminoAcid', 'vGeneName', 'jGeneName', 'patient', 'count (templates/reads)', 'sample']]
+        dfs.append(df)
+
 
+    # Concatenate all the dataframes into one
+    df_combined = pd.concat(dfs)
 
-# Concatenate all the dataframes into one
-df_combined = pd.concat(dfs)
+    # Rename columns as required
+    df_combined = df_combined.rename(columns={
+        'aminoAcid': 'CDR3b',
+        'vGeneName': 'TRBV',
+        'jGeneName': 'TRBJ',
+        'count (templates/reads)': 'counts'
+    })
+    df_combined = df_combined[df_combined['CDR3b'].notna()]
 
-# Rename columns as required
-df_combined = df_combined.rename(columns={
-    'aminoAcid': 'CDR3b',
-    'vGeneName': 'TRBV',
-    'jGeneName': 'TRBJ',
-    'count (templates/reads)': 'counts'
-})
-df_combined = df_combined[df_combined['CDR3b'].notna()]
+    df_combined.to_csv(f"concatenated_cdr3.txt", sep="\t", index=False, header=True)
 
-df_combined.to_csv(f"concatenated_cdr3.txt", sep="\t", index=False, header=True)
+if __name__ == "__main__":
+    main()

bin/sample_calc.py

@@ -13,33 +13,6 @@
 from scipy.stats import entropy
 import numpy as np
 import csv
-import os
-
-# initialize parser
-parser = argparse.ArgumentParser(description='Calculate clonality of a TCR repertoire')
-
-# add arguments
-parser.add_argument('-s', '--sample_meta', 
-                    metavar='sample_meta', 
-                    type=str, 
-                    help='sample metadata passed in through samples CSV file')
-parser.add_argument('-c', '--count_table', 
-                    metavar='count_table', 
-                    type=argparse.FileType('r'), 
-                    help='counts file in TSV format')
-
-args = parser.parse_args() 
-
-## convert metadata to list
-s = args.sample_meta
-sample_meta = args.sample_meta[1:-1].split(', ')
-# print('sample_meta looks like this: ' + str(sample_meta))
-
-# Read in the counts file
-counts = pd.read_csv(args.count_table, sep='\t', header=0)
-counts = counts.rename(columns={'count (templates/reads)': 'read_count', 'frequencyCount (%)': 'frequency'})
-# print('counts columns: \n')
-# print(counts.columns)
 
 def calc_sample_stats(sample_meta, counts):
     """Calculate sample level statistics of TCR repertoire."""
@@ -142,4 +115,31 @@ def calc_sample_stats(sample_meta, counts):
     # with open('gene_usage_' + str(metadata[1] + '_' + str(metadata[2] + '_' + str(metadata[3]))) + '.pkl', 'wb') as f:
     #     pickle.dump(gene_usage, f)
 
-calc_sample_stats(sample_meta, counts)
+def main():
+    # initialize parser
+    parser = argparse.ArgumentParser(description='Calculate clonality of a TCR repertoire')
+
+    # add arguments
+    parser.add_argument('-s', '--sample_meta', 
+                        metavar='sample_meta', 
+                        type=str, 
+                        help='sample metadata passed in through samples CSV file')
+    parser.add_argument('-c', '--count_table', 
+                        metavar='count_table', 
+                        type=argparse.FileType('r'), 
+                        help='counts file in TSV format')
+
+    args = parser.parse_args() 
+
+    ## convert metadata to list
+    s = args.sample_meta
+    sample_meta = args.sample_meta[1:-1].split(', ')
+
+    # Read in the counts file
+    counts = pd.read_csv(args.count_table, sep='\t', header=0)
+    counts = counts.rename(columns={'count (templates/reads)': 'read_count', 'frequencyCount (%)': 'frequency'})
+
+    calc_sample_stats(sample_meta, counts)
+
+if __name__ == "__main__":
+    main()

bin/samplesheet.py

@@ -1,24 +1,9 @@
 #!/usr/bin/env python
 
 import argparse
-import os
 import pandas as pd
 
-# initialize parser
-parser = argparse.ArgumentParser()
-parser.add_argument('-s', '--samplesheet', 
-                    metavar='samplesheet', 
-                    type=str, 
-                    help='sample metadata passed in through samples CSV file')
-
-parser.add_argument('-d', '--data_dir',
-                    metavar='data_dir',
-                    type=str,
-                    help='path to data directory')
-
-args = parser.parse_args()
-
-#do any processing of the samplesheet here
+# do any processing of the samplesheet here
 def samplesheet(samplesheet, data_dir):
     ss = pd.read_csv(samplesheet, sep=',')
     ss.to_csv('samplesheet_utf8.csv', index=False, encoding='utf-8-sig')
@@ -28,5 +13,22 @@ def samplesheet(samplesheet, data_dir):
 
     print(ss.head())
 
-samplesheet(args.samplesheet, args.data_dir)
-    
+def main():
+    # initialize parser
+    parser = argparse.ArgumentParser()
+    parser.add_argument('-s', '--samplesheet', 
+                        metavar='samplesheet', 
+                        type=str, 
+                        help='sample metadata passed in through samples CSV file')
+
+    parser.add_argument('-d', '--data_dir',
+                        metavar='data_dir',
+                        type=str,
+                        help='path to data directory')
+
+    args = parser.parse_args()
+
+    samplesheet(args.samplesheet, args.data_dir)
+
+if __name__ == "__main__":
+    main()